第一种:(推荐)只保存和加载模型参数 保存模型 torch.save(the_model.state_dict(), PATH) 加载模型参数 the_model = TheModelClass(*args, **kwargs) the_model.load_state_dict(torch.load(PATH)) 第二种:保存和加载整个模型 保存模型 torch.save(the_model, PATH) 加载模型 the_model = torch.load(PATH)