溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
PyTorch中train()方法的作用是什么，相信很多沒有經驗的人對此束手無策，為此本文總結了問題出現的原因和解決方法，通過這篇文章希望你能解決這個問題。
pytorch可以給我們提供兩種方式來 切換訓練和評估(推斷)的模式。分別是： model.train()和model.eval() dropout在訓練的時候起作用， 在推斷的時候被繞過不起作用， 或者等價地，將其概率置為零。 batch normalization 也和dropout一樣， 有兩種模式，分別對應訓練和推斷，分別是: model.eval()和model.train().
以下是原文截圖:
dropout:
batch normalization:
PyTorch官方API截圖:
train(mode=True):
eval():
看完上述內容，你們掌握PyTorch中train()方法的作用是什么的方法了嗎？如果還想學到更多技能或想了解更多相關內容，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
