PyTorch中的train()方法和eval()方法的做用和區別

pytorch能夠給咱們提供兩種方式來
切換訓練和評估(推斷)的模式。分別是：
model.train()和model.eval()


dropout在訓練的時候起做用，
在推斷的時候被繞過不起做用，
或者等價地，將其機率置爲零。

batch normalization 也和dropout同樣，
有兩種模式，分別對應訓練和推斷，分別是:
model.eval()和model.train().

如下是原文截圖:

dropout:

batch normalization:

PyTorch官方API截圖:
train(mode=True):

eval():

本文同步分享在 博客「敲代碼的小風」（CSDN）。
若有侵權，請聯繫 support@oschina.cn 刪除。
本文參與「OSC源創計劃」，歡迎正在閱讀的你也加入，一塊兒分享。