torch model.load_state_dict *** AttributeError: ‘ModelName’ 对象没有属性 ‘copy’

社会演员多 pytorch 283

原文标题 ：torch model.load_state_dict *** AttributeError: ‘ModelName’ object has no attribute ‘copy’

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

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'

原文链接：https://stackoverflow.com//questions/71574676/torch-model-load-state-dict-attributeerror-modelname-object-has-no-attrib

回复

我来回复
  • joe32140的头像
    joe32140 评论

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

    torch.save({'model': model.state_dict(),
            'scaler': scaler,
            'encoder': label_encoder,
            'config': config_parameters},
            trainedmodelpath)
    2年前 0条评论

相关问题