model.train()將模型設置為訓練狀態,作用:使Dropout,batchnorm知道后有不同表現(具體參考Dropout,batchnorm源碼),只有這兩個關心True or False。
將模型設置為測試狀態有兩種方法:
1.model.train(mode=False)
2.model.eval() 相當於第一種方法
model.train()源碼:
model.eval() 源碼:
model.train()將模型設置為訓練狀態,作用:使Dropout,batchnorm知道后有不同表現(具體參考Dropout,batchnorm源碼),只有這兩個關心True or False。
將模型設置為測試狀態有兩種方法:
1.model.train(mode=False)
2.model.eval() 相當於第一種方法
model.train()源碼:
model.eval() 源碼:
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。