報錯信息: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)