pytorch中保存模型

在pytorch中有两种模型保存方式，如下所示：

# 第一种：只保留模型参数：
torch.save(model.state_dict(),path) # 保存
model.load_state_dict(torch.load(path))   # 加载（需提前定义模型）

# 第二种：保存整个模型：
torch.save(model,path)    # 保存模型
model = torch.load(path)          # 加载模型