首页torch.load() AttributeError: 'dict' object has no attribute 'eval'

torch.load() AttributeError: 'dict' object has no attribute 'eval'

时间: 2023-12-06 22:38:54 浏览: 1102

解决运行出现dict object has no attribute has_key问题

5星 · 资源好评率100%

这个问题是py2和py3兼容性的问题在py2中，判断key是否属于dict的写法可以是： d={'name':'abc','location':'BeiJing'} if d.has_key('location'): print(d['location']) 在py3中，判断key是否属于字典的写法可以是： d={'name':'abc','location':'BeiJing'} if 'location' in d: print(d['location']) 补充知识：快速解决出现class object has no attribute ‘ functiong’ or ‘var

根据提供的引用内容，可以看出出现这个错误的原因是在调用`load_state_dict()`函数后，将返回的`_IncompatibleKeys`对象赋值给了模型，导致模型失去了原有的属性和方法。因此，正确的做法是直接在模型上调用`load_state_dict()`函数，而不是将其返回的对象赋值给模型。以下是修改后的代码示例： ```python # 保存的权重文件 resNet50 = resnet.resnet50() resNet50.load_state_dict(torch.load(resNet_path)['state_dict']) resNet50.eval() ```

阅读全文