pytorch模型的保存与加载 保存和加载整个模型仅保存和加载模型参数(推荐使用,需要提前手动构建模型) 保存和加载整个模型 torch.save(model_object, 'model.pkl') model = torch.load('model.pkl') 仅保存和加载模型参数(推荐使用,需要提前手动构建模型) torch.save(model_object.state_dict(), 'params.pkl') model_object.load_state_dict(torch.load('params.pkl'))