【深度学习】Pytorch中的模型的存储和加载

模型的存储

假设模型对象名称是model

model = Net()
#保存方法1
torch.save(model,"model_1.pth")

#方法2
torch.save(model.state_dict(),"model_2.pth")

模型的加载

#方法1 对应上面的方法1
model = Net()
model = torch.load("model_1.pth")

#方法2 对应上面的方法2
model = Net()
model.load_stat_dict(torch.load("model_2.pth"))

需要注意的是，不论是哪种加载方法，都需要先加载模型对象，不然是不可以进行模型的加载的