From 5b0755be21392dc55199485c753d6fc4390d5870 Mon Sep 17 00:00:00 2001
From: SanghyunPark01 <pash0302@naver.com>
Date: Sun, 19 Nov 2023 23:18:02 +0900
Subject: [PATCH] [ADD]load_model_rapidly

---
 src/Extractors/HFNetRTModel.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/Extractors/HFNetRTModel.cc b/src/Extractors/HFNetRTModel.cc
index 0e288ea..4611dc8 100644
--- a/src/Extractors/HFNetRTModel.cc
+++ b/src/Extractors/HFNetRTModel.cc
@@ -207,6 +207,8 @@ void HFNetRTModel::GetGlobalDescriptorFromTensor(const RTTensor &tDescriptors, c
 
 bool HFNetRTModel::LoadHFNetTRModel(void)
 {
+    if(LoadEngineFromFile(DecideEigenFileName(mStrTRModelDir, mMode, mInputShape)))return true;
+
     auto builder = unique_ptr<IBuilder>(createInferBuilder(mLogger));
     if (!builder) return false;
 
@@ -237,7 +239,7 @@ bool HFNetRTModel::LoadHFNetTRModel(void)
     if (mLogger.level >= ILogger::Severity::kINFO) PrintInputAndOutputsInfo(network);
 
     // Save Engine
-    // SaveEngineToFile(DecideEigenFileName(mStrTRModelDir, mMode, mInputShape), serializedEngine);
+    SaveEngineToFile(DecideEigenFileName(mStrTRModelDir, mMode, mInputShape), serializedEngine);
     
     unique_ptr<IRuntime> runtime{createInferRuntime(mLogger)};
     if (!runtime) return false;