P28_完整的模型训练套路(三)
本节进行训练细节总结
打开官网:torch.nn.Module,可以查看train mode 和eval mode
.train()和.eval()只对特定层如Dropout和BatchNorm层起作用,但是最好还是加上;
即在训练开始前加上.train(),在测试开始前加上.eval():
点击查看代码
#8.4训练步骤开始:
dyl.train()
......
#9.测试步骤开始:
dyl.eval()
本节进行训练细节总结
打开官网:torch.nn.Module,可以查看train mode 和eval mode
.train()和.eval()只对特定层如Dropout和BatchNorm层起作用,但是最好还是加上;
即在训练开始前加上.train(),在测试开始前加上.eval():
#8.4训练步骤开始:
dyl.train()
......
#9.测试步骤开始:
dyl.eval()