add函数 pytorch_Pytorch的高级训练，数据增强和实用程序 - pytorch中文网

v0.1.3 JUST RELEASED - 包含显着的改进，错误修复和额外的支持。从版本中获取，或者拉主分支。

这个包提供了几件事情：

具有回调，约束和正则化程序的类似Keras的训练的高级模块。

全面的数据增加，变换，采样和加载

实用张量和变量函数，所以你不需要经常使用numpy

有任何功能要求吗？提交问题！我会做到这一点。具体来说，任何数据增加，数据加载或采样功能。

想贡献？检查 标签为[贡献欢迎] 的问题页面。

ModuleTrainer

该ModuleTrainer课程提供了一个高级训练界面，可以提供回调，限制，初始化，正则化等等，从而抽取训练循环。

例：

from torchsample.modules import ModuleTrainer

# Define your model EXACTLY as normal

class Network(nn.Module):

def __init__(self):

super(Network, self).__init__()

self.conv1 = nn.Conv2d(1, 32, kernel_size=3)

self.conv2 = nn.Conv2d(32, 64, kernel_size=3)

self.fc1 = nn.Linear(1600, 128)

self.fc2 = nn.Linear(128, 10)

def forward(self, x):

x = F.r