has no attribute 'to'
时间: 2023-10-10 07:15:33 浏览: 114
Python 从attribute到property详解
在引用中提到了一个错误,即AttributeError: ‘_IncompatibleKeys’ object has no attribute ‘to’。这个错误发生在你尝试将模型加载到设备上时。根据引用中的解决方法,load_state_dict()方法不会返回一个模型,而是一个_IncompatibleKeys对象,所以你不能直接将其赋值给model。因此,你需要在加载模型后重新赋值给model,并且再次使用to()方法将其移动到设备上。
下面是一个示例解决方法:
model.load_state_dict(torch.load("bestmodel.pth"))
model = model.to(device)
model.eval()
阅读全文