摘要: model.train():在使用pytorch构建神经网络的时候,训练过程中会在程序上方添加一句model.train(),作用是启用batch normalization和drop out。 model.eval():测试过程中会使用model.eval(),这时神经网络会沿用batch nor 阅读全文
posted @ 2022-11-26 17:47 wieneralan 阅读(83) 评论(0) 推荐(0) 编辑