pytorch模型存储的两种方式

1.保存整个网络结构信息和模型参数信息:torch.save(model_object, './model.pth')直接加载即可使用：model = torch.load('./model.pth'