pytorch模型需要重新在npu上训练才能使用吗
收藏回复举报
pytorch模型需要重新在npu上训练才能使用吗
新人帖
发表于2023-01-29 14:15:16
0 查看

已经有利用gpu训练好的torch模型,.pth文件。现在想推理时使用晟腾芯片和鲲鹏这套组合。

查阅相关文档发现的都是 先把模型迁移到npu下面,再重新训练,标定精度等等。

也就是说gpu下训练好的模型文件不能直接在npu下load再eval推理吗?

我要发帖子