# 保存和加载整个模型
torch.save(model_object, 'path/model.pkl')
model = torch.load('path/model.pkl')
# 仅保存和加载模型参数(推荐使用)
torch.save(model_object.state_dict(), 'path/params.pkl')
model_object.load_state_dict(torch.load('path/params.pkl'))
# 保存和加载整个模型
torch.save(model_object, 'path/model.pkl')
model = torch.load('path/model.pkl')
# 仅保存和加载模型参数(推荐使用)
torch.save(model_object.state_dict(), 'path/params.pkl')
model_object.load_state_dict(torch.load('path/params.pkl'))