tensortx模型部署问题解决collections.OrderedDict‘ object has no attribute ‘to‘ ,‘eval‘

最新推荐文章于 2024-08-28 21:34:07 发布

什么是快了星球

最新推荐文章于 2024-08-28 21:34:07 发布

阅读量4.9k

点赞数 3

CC 4.0 BY-SA版权

分类专栏：深度学习环境文章标签：深度学习

本文链接：https://blog.csdn.net/qq_39076073/article/details/118708422

在PyTorch 1.6.0环境下，保存模型时使用`torch.save(model.state_dict(), path)`会导致在加载模型进行预测时出现'collections.OrderedDict' object has no attribute 'to'和'eval'的错误。正确的方法是使用`torch.save(model, path)`来保存整个模型，或者在模型训练中仅保存权重参数，避免使用`model.state_dict()`，因为这不会保存模型的完整状态。" 115376396,10768653,全排列算法详解,"['算法', '数据结构', '递归', '编程']

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

‘collections.OrderedDict’ object has no attribute ‘to’

this is my main code,but I don't know how to fix the problem?

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
model = torch.load('./checkpoints/fcn_model_5.pth')