diff --git a/lite/load_and_run/src/models/model_lite.cpp b/lite/load_and_run/src/models/model_lite.cpp index 656e1d9ec514baeed1634c9f9c7a36c58ac7fb72..64d6833b9fbae0c1caf1f064758c7f2c9e242b4c 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 ba79ed28915ea77d5bf232b93f878a781147f952..1df00b29c52bffe4951b2027558eae7eeb174596 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;