当前位置: 首页 > news >正文

P28_完整的模型训练套路(三)

P28_完整的模型训练套路(三)

本节进行训练细节总结

打开官网:torch.nn.Module,可以查看train mode 和eval mode

.train()和.eval()只对特定层如Dropout和BatchNorm层起作用,但是最好还是加上;

即在训练开始前加上.train(),在测试开始前加上.eval():

点击查看代码
#8.4训练步骤开始:
dyl.train()
......#9.测试步骤开始:
dyl.eval()