Pytorch基础——模型的保存和加载

保存

• 保存整个模型的结构信息和参数信息，对象model

torch.save(model, '路径'）

• 保存参数、保存对象是模型的状态model.state_dict

torch.save(model.state_dict, '路径'）

加载

• 加载完整的模型结构

load_model = torch.load('路径'）

• 加载模型参数信息、需要先导入模型的结构，然后

model.load_state_dict(torch.load('路径'))