pytorch 在加載模型時出現的錯誤

在訓練完後，在另一個.py中想導入模型進行使用。導入之後出現這樣的錯誤： pytorch AttributeError: Can’t get attribute ‘Net1’ on <module ‘main’> 網上有很多原因， 有一部分是說多線程。 但經過查看文檔，這種情況不屬於多線程的問題。 是因爲在保存模型時pytorch保存整個模型的序列化數據綁定到特定類以及保存模型時使用的確切目錄結構

>>阅读原文<<