首页 > 其他 > 详细

关于Pytorch中model.train()

时间:2021-03-11 22:23:16      阅读:229      评论:0      收藏:0      [点我收藏+]

model.train()将模型设置为训练状态,作用:使Dropout,batchnorm知道后有不同表现(具体参考Dropout,batchnorm源码),只有这两个关心True or False。
将模型设置为测试状态有两种方法:
1.model.train(mode=False)
2.model.eval() 相当于第一种方法

model.train()源码:
技术分享图片

model.eval() 源码:
技术分享图片

关于Pytorch中model.train()

原文:https://www.cnblogs.com/wmc258/p/14520098.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!