Aaapirl 2023-03-24 10:18 采纳率: 9.1%
浏览 9

用pytorch1.7训练的文件.pth在pytorch1.2上无法进行预测。

用pytorch1.7训练的文件.pth在pytorch1.2上无法进行预测。
请问应该怎么改

  • 写回答

1条回答 默认 最新

  • 爱晚乏客游 2023-03-24 11:17
    关注

    应该是某些算子之类的不支持问题,你的能否正确读取模型?
    模型权重能读取的话,将网络结构重新用1.2的版本实现下,然后读取权重赋值下试试看。
    这路子不行的话,就得换方案了,比如转成onnx然后用其他方式推理

    评论

报告相同问题?

问题事件

  • 创建了问题 3月24日