原创 PyTorch加載自己訓練好模型時一個坑

抽個空,趕緊把這個錯誤記一下,怕忘記了. 前幾天,很開心地訓練了一個較好的模型,說讓從中抽出一個測試的API出來.於是,吭哧吭哧地整了一天,發現怎麼同一個模型,同一張圖片,結果總是不同呢? 懵逼中… … 幾經周折,發現了問題所在