我們在訓練時如果使用了BN層和Dropout層，我們需要對model進行標識：
model.train():在訓練時使用BN層和Dropout層，對模型進行更改。
model.eval():在評價時將BN層和Dropout層凍結，這兩個操作不會對模型進行更改。
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。
猜您在找
pytorch 使用的時候的 model.train() 和 model.eval()
Pytorch學習筆記11----model.train()與model.eval()的用法、Dropout原理、relu,sigmiod,tanh激活函數、nn.Linear淺析、輸出整個tensor的方法
PyTorch中eval模式和train模式
[pytorch筆記] torch.nn vs torch.nn.functional; model.eval() vs torch.no_grad(); nn.Sequential() vs nn.moduleList
Pytorch的net.train 和 net.eval的使用
pytorch中model.modules()和model.children()的區別
eval和new Function的區別
execScript與eval區別
caffe的model和prototxt轉pytorch的model
sklearn.model_selection.train_test_split 用法