参考链接

https://blog.csdn.net/irober/article/details/115144522?utm_medium=distribute.pc_relevant.none-task-blog-baidujs_title-0&spm=1001.2101.3001.4242

torch版本问题

import torch
state_dict = torch.load("./model_00050.pt")#加载原来的模型  在torch=1.6时加载
torch.save(state_dict, "./00050.pt", _use_new_zipfile_serialization=False)#不是zip 

再加载save后的模型即可

相关文章:

  • 2022-12-23
  • 2021-11-09
  • 2022-12-23
  • 2021-04-22
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-02-05
猜你喜欢
  • 2022-12-23
  • 2021-06-27
  • 2022-12-23
  • 2021-10-05
  • 2021-11-02
  • 2022-12-23
  • 2021-12-10
相关资源
相似解决方案