问题描述:想要使用训练好的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版本的要求)