提交 5cc06f8e 编写于 作者: Z zhangyang0701

support lod_mode for FPGA track

上级 18ef6706
...@@ -41,10 +41,12 @@ bool PaddleMobilePredictor<Device, T>::Init(const PaddleMobileConfig &config) { ...@@ -41,10 +41,12 @@ bool PaddleMobilePredictor<Device, T>::Init(const PaddleMobileConfig &config) {
config.memory_pack.combined_params_buf); config.memory_pack.combined_params_buf);
} else if (!config.model_dir.empty()) { } else if (!config.model_dir.empty()) {
paddle_mobile_->Load(config.model_dir, config.optimize, paddle_mobile_->Load(config.model_dir, config.optimize,
config.quantification, config.batch_size); config.quantification, config.batch_size,
config.lod_mode);
} else if (!config.prog_file.empty() && !config.param_file.empty()) { } else if (!config.prog_file.empty() && !config.param_file.empty()) {
paddle_mobile_->Load(config.prog_file, config.param_file, config.optimize, paddle_mobile_->Load(config.prog_file, config.param_file, config.optimize,
config.quantification, config.batch_size); config.quantification, config.batch_size,
config.lod_mode);
} else { } else {
LOG(kLOG_ERROR) << "fail to load inference model!"; LOG(kLOG_ERROR) << "fail to load inference model!";
return false; return false;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册