Samar-blog

导航

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

本节进行训练细节总结

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

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

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

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

#9.测试步骤开始:
dyl.eval()

posted on 2025-11-24 22:27  风居住的街道DYL  阅读(0)  评论(0)    收藏  举报