torch model.load_state_dict *** AttributeError: 'ModelName' object has no attribute 'copy'

Question

我以前保存过这样的模型：

trainedmodelpath = "model.th"
torch.save({'model': model,
            'scaler': scaler,
            'encoder': label_encoder,
            'config': config_parameters},
            trainedmodelpath)

但是当我尝试像这样加载它时：

PreviousModelPath = "model.th"
TorchLoadedState = torch.load(PreviousModelPath)
TorchLoadedState_Model = TorchLoadedState['model']
model.load_state_dict(TorchLoadedState_Model)

我收到这个错误

*** AttributeError: 'ModelName' object has no attribute 'copy'

Answer 1

使用 model.state_dict() 保存模型权重：

torch.save({'model': model.state_dict(),
            'scaler': scaler,
            'encoder': label_encoder,
            'config': config_parameters},
            trainedmodelpath)

torch model.load_state_dict *** AttributeError: 'ModelName' object has no attribute 'copy'

torch model.load_state_dict *** AttributeError: 'ModelName' object has no attribute 'copy'

python

deep-learning

pytorch