Pytorch模型保存和加载

保存模型：

torch.save(model, 'model.pth')

加载模型：

model = torch.load('model.pth')