torch版本导致的文件保存不一致

问题描述:想要使用训练好的pt模型继续训练,在加载模型时出现以下提示
在这里插入图片描述
原因:1.6之后pytorch默认使用zip文件格式来保存权重文件,导致这些权重文件无法直接被1.5及以下的pytorch加载。如torch1.2无法使用torch1.6生成的pt文件
解决1:在1.6版本的train.py添加_use_new_zipfile_serialization=False后训练得到的模型可用于1.2版本中(使用torch.save 进行保存为非zip格式)

torch.save(model.state_dict(), model_cp,_use_new_zipfile_serialization=False)  # 训练所有数据后,保存网络的参数

解决2:将1.2升级为1.6

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.6.0 torchvision==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

升级时间长,而且受之前安装环境的影响,并不一定安装成功

  • 又可能会出现一些错误,比如RuntimeError: No such operator torchvision::nms 问题,
    原因:pytorch和torchvision的版本不匹配
    解决方法:升级torchvision版本,或者重新安装pytorch(其实就是一定要按照同一个项目的配置配置环境,不然就容易出问题,即先看requirment对torch+torchvision版本的要求)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值