用pytorch1.7训练的文件.pth在pytorch1.2上无法进行预测。
请问应该怎么改
用pytorch1.7训练的文件.pth在pytorch1.2上无法进行预测。
- 写回答
- 好问题 0 提建议
- 关注问题
- 邀请回答
-
1条回答 默认 最新
爱晚乏客游 2023-03-24 11:17关注应该是某些算子之类的不支持问题,你的能否正确读取模型?
模型权重能读取的话,将网络结构重新用1.2的版本实现下,然后读取权重赋值下试试看。
这路子不行的话,就得换方案了,比如转成onnx然后用其他方式推理解决 无用评论 打赏 举报