pytorch:model.train和model.eval用法和區別

使用PyTorch進行訓練和測試時必定注意要把實例化的model指定train/eval，eval（）時，框架會自動把BN和DropOut固定住，不會取平均，而是用訓練好的值，否則的話，一旦test的batch_size太小，很容易就會被BN層致使生成圖片顏色失真極大！！！！！！web Class Inpaint_Network() ...... Model = Inpaint_Nerwoek

>>阅读原文<<