Pytorch模型保存和加载

保存模型:

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

加载模型:

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

你可能感兴趣的:(深度学习,python,机器学习,tensorflow,java)