pytorch中关于模型的保存分为三种形式

第一种：保存整个模型

##模型保存
torch.save(model,"PATH")
##模型加载
model = torch.load("PATH")
##设置模型进行测试模式
model.eval()

第二种：仅保存模型的各项参数

##其中model.state_dict()表示模型的各项参数
torch.save(model.state_dict(), "PATH")
##加载模型，注意，这种方式首先要将模型的原型写出后，才能加载模型的各项指标参数
model = TheModelClass(*args,*kwargs)
model.load_state_dict(torch.load("PATH"))
model.eval()

第三种：TorchScript形式

model_scripted = torch.jit.script(model) # Export toTorchScript
model_scripted.save("model_scripted.pt") # Save
model = torch.jit.load("model _scripted.pt")
model.eval()