Pytorch的net.train 和 net.eval的使用

在訓練模型時會在前面加上:python model.train() 在測試模型時在前面使用:web model.eval() 同時發現,若是不寫這兩個程序也能夠運行,這是由於這兩個方法是針對在網絡訓練和測試時採用不一樣方式的狀況,好比Batch Normalization 和 Dropout。網絡 訓練時是正對每一個min-batch的,可是在測試中每每是針對單張圖片,即不存在min-batch的
相關文章
相關標籤/搜索