pytorch掉坑記錄：model.eval的作用

 

訓練完train_datasets之后，model要來測試樣本了。在model(test_datasets)之前，需要加上model.eval(). 否則的話，有輸入數據，即使不訓練，它也會改變權值。這是model中含有batch normalization層所帶來的的性質。
在做one classification的時候，訓練集和測試集的樣本分布是不一樣的，尤其需要注意這一點。