pytorch保存和加載模型的兩種方式

"""
pytorch的模型保存和加載的兩種方式：
方式1：只保存和恢復模型中的參數，使用這種方法，咱們須要本身導入模型的結構信息
保存：
torch.save(model.state_dict(), PATH)------torch.save(vgg.state_dict(), 'model.pth')
恢復：
model = net()---須要本身導入模型的結構
model.load_state_dict(torch.load('model.pth'))

方式2：使用這種方法，將會保存模型的參數和結構信息
保存：
torch.save(model, PATH)
恢復：
torch.load(PATH)
這種方式獲得的模型佔用空間很是大，不推薦使用！！