火炬样本:Pytorch的高级培训数据增强和实用程序 源码
Pytorch的高级培训,数据增强和实用程序 -包含重大改进,错误修复和其他支持。 从发行版中获取它,或拉到master分支。 该软件包提供了一些东西: 一个具有回调,约束和正则化器的类似于Keras训练的高级模块。 全面的数据扩充,转换,采样和加载 实用张量和变量函数,因此您不需要经常使用numpy 有任何功能要求吗? 提交问题! 我会实现的。 具体而言,任何数据扩充,数据加载或采样功能。 想要贡献? 在查看那些标有[欢迎捐款]的内容。 培训师 ModuleTrainer类提供了高级培训接口,该接口抽象了培训循环,同时提供了回调,约束,初始化器,正则化器等。 例: from t
用户评论