【pytorch】model.eval()和model.train()

eval() self.eval()和self.train(False)等價 eval()在測試之前加，否則有輸入數據即使不訓練，它也會改變權值 pytorch會自己把BatchNormalization和DropOut固定住，不會取平均，而是用訓練好的值。 train() model.train()讓model變成訓練模式

>>阅读原文<<