zl程序教程

您现在的位置是:首页 >  后端

当前栏目

【深度学习torch——error】——“xxx.pt is a zip archive(did you mean to use torch.jit.load()?)

zip学习 Error to 深度 is use load
2023-09-11 14:21:45 时间

这个问题是在进行权重文件加载进行预测的时候发生的,原因其实就是torch版本不对

我是用的工作站训练使用的是torch1.7.0,然后用自己的电脑进行预测,就报错了,原因就是自己的电脑是torch1.2.0版本的

因为在1.6版本以上的模型改变了权重文件的读取格式,所以尽量版本之间不要差得太大