PyTorch深度学习实战

使用PyTorch进行深度学习实战

使用PyTorch进行深度学习实战，对我来说一般编写四个文件来完成深度学习项目：

• main文件
• engine文件
• model文件
• utils文件

main文件

main文件主要进行参数的解析argparse，链接其他三个文件，将数据x,y输入模型得到输出loss，保存模型等，在main文件里主要进行以下几个过程：

engine=trainer(lr,decay,dim,....)
train_loss=engine.train(x,y)
test_loss=enginer.eval(x,y)

model文件

在model文件里主要进行网络结构部分的组织和定义，有一个大类，如下：

class Net1(nn.<