diff --git a/lite/load_and_run/src/models/model_lite.cpp b/lite/load_and_run/src/models/model_lite.cpp index 656e1d9e..64d6833b 100644 --- a/lite/load_and_run/src/models/model_lite.cpp +++ b/lite/load_and_run/src/models/model_lite.cpp @@ -13,6 +13,7 @@ ModelLite::ModelLite(const std::string& path) : model_path(path) { void ModelLite::load_model() { m_network = std::make_shared(config, IO); if (enable_layout_transform) { + LITE_WARN("enable layout transform while load model for lite"); lite::Runtime::enable_global_layout_transform(m_network); } if (share_model_mem) { diff --git a/lite/load_and_run/src/models/model_lite.h b/lite/load_and_run/src/models/model_lite.h index ba79ed28..1df00b29 100644 --- a/lite/load_and_run/src/models/model_lite.h +++ b/lite/load_and_run/src/models/model_lite.h @@ -60,8 +60,8 @@ public: std::vector get_model_data() override; private: - bool share_model_mem; - bool enable_layout_transform; + bool share_model_mem = false; + bool enable_layout_transform = false; std::string model_path; DataParser parser;