无法在 Pytorch 版本 1.5.1 中加载在 Pytotch 版本 1.6.0 上训练的模型

Question

我最近在1.6.0本地机器上更新了我的 Pytorch 版本以使用他们的混合精度训练，从那时起我遇到了这个问题，我尝试了这里提到的解决方案，但它仍然抛出以下错误。

RuntimeError: version_ <= kMaxSupportedFileFormatVersion INTERNAL ASSERT FAILED at /opt/conda/conda-bld/pytorch_1591914880026/work/caffe2/serialize/inline_container.cc:132, please report a bug to PyTorch. Attempted to read a PyTorch file with version 4, but the maximum supported version for reading is 3. Your PyTorch installation may be too old.

重现链接：https://www.kaggle.com/rohitsingh9990/error-reducing-code ?scriptVersionId=37468859

任何帮助将不胜感激，提前致谢。

Answer 1

在torch.__version__ == 1.6.0：

torch.save(model_.state_dict(), 'best_model.pth.tar', use_new_zipfile_serialization=False)

然后在torch.__version__ == 1.5.1：

torch.load('best_model.pth.tar',map_location='cpu')