|
@@ -13,6 +13,7 @@ ModelLite::ModelLite(const std::string& path) : model_path(path) { |
|
|
void ModelLite::load_model() { |
|
|
void ModelLite::load_model() { |
|
|
m_network = std::make_shared<lite::Network>(config, IO); |
|
|
m_network = std::make_shared<lite::Network>(config, IO); |
|
|
if (enable_layout_transform) { |
|
|
if (enable_layout_transform) { |
|
|
|
|
|
LITE_WARN("enable layout transform while load model for lite"); |
|
|
lite::Runtime::enable_global_layout_transform(m_network); |
|
|
lite::Runtime::enable_global_layout_transform(m_network); |
|
|
} |
|
|
} |
|
|
if (share_model_mem) { |
|
|
if (share_model_mem) { |
|
|