PyTorch 模型的保存和加載

1. Pytorch 加載完整模型的參數 保存加載整個模型 # 保存整個模型 torch.save (model_object, 'model.pk1') # 加載整個模型 model = torch.load('model.pkl') 保存模型的參數 (推薦使用) # 模型參數保存 torch.save (model_object.state_dict(), 'params.pk1') # 保

>>阅读原文<<