提交 2c1feaa5 编写于 作者: L liuruilong

format files

上级 7faca1ad
......@@ -75,7 +75,7 @@ class Executor {
int batch_size_;
bool use_optimize_;
bool lod_mode_;
PaddleMobileConfigInternal config_ = PaddleMobileConfigInternal();
PaddleMobileConfigInternal config_;
Program<Device> program_;
std::shared_ptr<ProgramDesc> program_desc_;
typedef std::shared_ptr<OperatorBase<Device>> OperatorBasePtr;
......
......@@ -45,8 +45,8 @@ PMStatus PaddleMobile<Device, T>::Load(const std::string &dirname,
if (executor_.get() == nullptr) {
executor_ = std::make_shared<framework::Executor<Device, T>>(
loader_->Load(dirname, optimize, quantification), batch_size, optimize,
lod_mode);
loader_->Load(dirname, optimize, quantification), config_, batch_size,
optimize, lod_mode);
} else {
LOG(kLOG_INFO) << "executor inited";
}
......@@ -67,7 +67,7 @@ PMStatus PaddleMobile<Device, T>::Load(const std::string &model_path,
if (executor_.get() == nullptr) {
executor_ = std::make_shared<framework::Executor<Device, T>>(
loader_->Load(model_path, para_path, optimize, quantification),
loader_->Load(model_path, para_path, optimize, quantification), config_,
batch_size, optimize, lod_mode);
} else {
LOG(kLOG_INFO) << "executor inited";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册