import torch
save_model_path='model.pkl'
#保存模型
torch.save(model, save_model_path)
#加载模型
model = torch.load(save_model_path)
import torch
save_model_path='model.pkl'
#保存模型
torch.save(model, save_model_path)
#加载模型
model = torch.load(save_model_path)