在pytorch中有两种模型保存方式,如下所示:
# 第一种:只保留模型参数: torch.save(model.state_dict(),path) # 保存 model.load_state_dict(path) # 加载(需提前定义模型) # 第二种:保存整个模型: torch.save(model,path) # 保存模型 model = torch.load(path) # 加载模型