PyTorch 模型的保存和加载

1. Pytorch 加载完整模型的参数 保存加载整个模型 # 保存整个模型 torch.save (model_object, 'model.pk1') # 加载整个模型 model = torch.load('model.pkl') 保存模型的参数 (推荐使用) # 模型参数保存 torch.save (model_object.state_dict(), 'params.pk1') # 保

>>阅读原文<<