模型訓練與測試

model.train() model.eval() 一般在模型訓練和評價的時候會加上這兩句，主要是針對model 在訓練時和評價時不同的 Batch Normalization 和 Dropout 方法模式。 eval()時，pytorch會自動把BN和DropOut固定住，不會取平均，而是用訓練好的值。不然的話，一旦test的batch_size過小，很容易就會被BN層導致生成圖片顏色失真極大

>>阅读原文<<