Pytorch学习笔记02 - 20210613

模型的保存与加载

一、保存

(一)、torch.save() 和 torch.load()

1. 保存整个模型：

# save
torch.save(moedl,'save_model.pth')
# 调用
model = torch.load('save_model.pth')

2. 仅保存模型参数

# 保存
torch.save(the_model.state_dict(), PATH='model_parameters.pth')		#只保存模型权重参数，不保存模型结构
# 调用
the_model = Model(*args, **kwargs)