两种方法：保存和加载参数 和 保存加载整个模型
保存和加载参数
#保存
torch.save(model.state_dict,PATH) #PATH推荐格式为.pt
#加载
model=TheModelClass(*args, **kwargs )
model.load_state_dict(torch.load(PATH) )保存加载整个模型
#保存
torch.save(model,PATH)
#加载
model = torch.load(PATH)标签：load,模型,torch,保存,pytorch,PATH,model,加载 From： https://www.cnblogs.com/CLGYPYJ/p/17610588.html