參考鏈接
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后的模型即可