Pytorch Pre-trained RESNET18モデル

Question

私は、訓練されたRESNET18モデルをパイロットで訓練し、保存しました。モデルのテストでは、異なるミニバッチサイズに対して異なる精度が与えられます。なぜ誰が知っていますか？

Answer 1

はい、そうです。 RESNETには、バッチ規格化レイヤが含まれています。評価時にこれらを修正する必要があります。それ以外の場合は、各バッチを処理した後に連続的に調整されているので、異なる精度が得られます。

は設定してみてください：評価の前に

model.eval() 

。トレーニングに戻る前に、model.train()に電話してください。