Backto PyTorch Index
模型保存
# 保存整个模型的结构信息和参数信息, 保存的对象是模型 model ,可以是pth方式,也可以是pkl的方式
torch.save(model, './model.pth') # 方式1
# 保存模型的参数, 保存的对象是模型的状态 model.state_dict()
torch.save(model.state_dict(), './model_state.pth'
Backto PyTorch Index
# 保存整个模型的结构信息和参数信息, 保存的对象是模型 model ,可以是pth方式,也可以是pkl的方式
torch.save(model, './model.pth') # 方式1
# 保存模型的参数, 保存的对象是模型的状态 model.state_dict()
torch.save(model.state_dict(), './model_state.pth'