Advertisement

pytorch 1.7训练保存的模型在1.4低版本无法加载:frame #63: <unknown function> + 0x1db3e0 (0x55ba98ddd3e0 in /data/user

阅读量:

使用pytorch 1.7及以上版本进行训练并保存的模型,在尝试于1.4版本中加载时会出现兼容性问题,并引发相应的错误提示。

复制代码
    torch.load('/home/user1/model_best_b.pth.tar')
    Traceback (most recent call last):
      File "/data/user1/pkgs/conda/envs/drc/lib/python3.7/site-packages/IPython/core/interactiveshell.py", line 3417, in run_code
    exec(code_obj, self.user_global_ns, self.user_ns)
      File "<ipython-input-10-13d633918c2f>", line 1, in <module>
    torch.load('/home/user1/pjs/fae/paper/ckpt/to_test/20201204175958/Arcface50_t4_bs50_bslr_0.001_fclr_0.01/model_best_bacc.pth.tar')
      File "/data/user1/pkgs/conda/envs/drc/lib/python3.7

全部评论 (0)

还没有任何评论哟~