pytorch model.eval()的作用
生活随笔
收集整理的這篇文章主要介紹了
pytorch model.eval()的作用
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
驗證模塊改成驗證模式。
本質上是將dropout作用去掉及不使用bn計算。這兩個僅僅在訓練時需要用到。
總結
以上是生活随笔為你收集整理的pytorch model.eval()的作用的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: pytorch 不同设备下保存和加载模型
- 下一篇: __getitem__的作用