关于Pytorch中model.train()


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

model.train()源码:

model.eval() 源码:


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM