这个问题是在进行权重文件加载进行预测的时候发生的,原因其实就是torch版本不对
我是用的工作站训练使用的是torch1.7.0,然后用自己的电脑进行预测,就报错了,原因就是自己的电脑是torch1.2.0版本的
因为在1.6版本以上的模型改变了权重文件的读取格式,所以尽量版本之间不要差得太大
这个问题是在进行权重文件加载进行预测的时候发生的,原因其实就是torch版本不对
我是用的工作站训练使用的是torch1.7.0,然后用自己的电脑进行预测,就报错了,原因就是自己的电脑是torch1.2.0版本的
因为在1.6版本以上的模型改变了权重文件的读取格式,所以尽量版本之间不要差得太大