From 19bf2db549d157dd361eab3ef190b9f736fdcdec Mon Sep 17 00:00:00 2001 From: Bo Zhou <2466956298@qq.com> Date: Tue, 31 Mar 2020 15:39:22 +0800 Subject: [PATCH] Update async_es_agent.cc --- deepes/src/paddle/async_es_agent.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepes/src/paddle/async_es_agent.cc b/deepes/src/paddle/async_es_agent.cc index 50fc2a0..b120eec 100644 --- a/deepes/src/paddle/async_es_agent.cc +++ b/deepes/src/paddle/async_es_agent.cc @@ -49,7 +49,7 @@ bool AsyncESAgent::_save() { model_name = "model_iter_id-"+ std::to_string(model_iter_id); std::string model_path = _config->async_es().model_warehouse() + "/" + model_name; LOG(INFO) << "[save]model_path: " << model_path; - _predictor->SaveOptimizedModel(model_path, paddle::lite_api::LiteModelType::kProtobuf); + _predictor->SaveOptimizedModel(model_path, LiteModelType::kProtobuf); // save config auto async_es = _config->mutable_async_es(); async_es->set_model_iter_id(model_iter_id); -- GitLab