AttributeError: 'Classifier' object has no attribute 'state_dic'
时间: 2023-10-30 20:02:18 浏览: 102
【Python】AttributeError: ‘AutoSchema’ object has no attribute ‘get_link’
5星 · 资源好评率100%
这个错误很可能是由于你的代码中存在拼写错误导致的。具体来说,AttributeError: 'Classifier' object has no attribute 'state_dic'错误表明在"Classifier"对象中找不到"state_dict"属性。根据你提供的代码引用,你创建了一个名为"classifier"的模型对象,但在加载模型时,你使用了错误的属性名。正确的属性名应该是"state_dict",而不是"state_dic"。
要解决这个问题,你需要将加载模型时的属性名从"state_dic"改为"state_dict"。具体来说,你可以使用以下代码来加载模型的状态字典:
state_dict_load = torch.load(path_state_dict)["state_dict"]
然后,你可以将加载的状态字典加载到你的模型对象中,例如:
classifier.load_state_dict(state_dict_load)
这样,你就能成功加载模型的状态字典并解决AttributeError: 'Classifier' object has no attribute 'state_dic'错误了。
阅读全文