Pytorch中的 model.train() 和 model.eval() 模式


model.train() :啟用 BatchNormalization 和 Dropout

model.eval() :不啟用 BatchNormalization 和 Dropout

 

參考:

https://pytorch.org/docs/stable/nn.html?highlight=module%20eval#torch.nn.Module.eval


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM