pytorch--報錯 AttributeError: ‘Net’ object has no attribute ‘copy’

報錯信息：AttributeError: ‘Net’ object has no attribute ‘copy’

 

分析：報錯是發生在加載預訓練模型時，很可能時兩種pytorch加載預訓練模型方式弄混了。

解決：按照下面思路加載預訓練模型就好。

1.保存加載state_dict方式（推薦）

保存：torch.save(model.state_dict(), PATH) # 推薦的文件后綴名是pt或pth

加載：model = TheModelClass(*args, **kwargs)

　　    model.load_state_dict(torch.load(PATH))

 2.保存加載整個模型

保存：torch.save(model, PATH)

加載：model = torch.load(PATH)