commit b88705d6c04d8b7b2fdffaf1501042b984e682cf
author Isabella Gottardi <isabella.gottardi@arm.com> Wed Apr 21 13:09:51 2021 +0100
committer kshitij-sisodia-arm <kshitij-sisodia-arm@review.mlplatform.org> Fri Apr 23 16:13:52 2021 +0000
tree 8d5415268fb5b46adc82f0c62de866210325d6dc
parent fcee088fe85f029711e9e190d7bd38b454fc6e6b

MLECO-1880: Make model zoo commit hash configurable for each model

Signed-off-by: Isabella Gottardi <isabella.gottardi@arm.com>
Change-Id: Ic7bcb0cbf334049ff89e9973e58fd0b3bf463892

source/use_case/img_class/usecase.cmake

diff --git a/source/use_case/img_class/usecase.cmake b/source/use_case/img_class/usecase.cmake
index 440eabe..384c58e 100644
--- a/source/use_case/img_class/usecase.cmake
+++ b/source/use_case/img_class/usecase.cmake
@@ -72,8 +72,9 @@
     # Download the default model
     set(ZOO_COMMON_SUBPATH      "models/image_classification/mobilenet_v2_1.0_224/tflite_uint8")
     set(ZOO_MODEL_SUBPATH       "${ZOO_COMMON_SUBPATH}/${MODEL_FILENAME}")
+    set(ZOO_MODEL_VERSION       "68b5fbc77ed28e67b2efc915997ea4477c1d9d5b")
 
-    download_file_from_modelzoo(${ZOO_MODEL_SUBPATH}    ${DEFAULT_MODEL_PATH})
+    download_file_from_modelzoo(${ZOO_MODEL_VERSION} ${ZOO_MODEL_SUBPATH} ${DEFAULT_MODEL_PATH})
 
     if (ETHOS_U55_ENABLED)
         message(STATUS
@@ -94,8 +95,8 @@
         set(${use_case}_TEST_OFM    ${MODEL_RESOURCES_DIR}/ofm0.npy CACHE FILEPATH
                                     "Input test vector for ${use_case}")
 
-        download_file_from_modelzoo(${ZOO_TEST_IFM_SUBPATH} ${${use_case}_TEST_IFM})
-        download_file_from_modelzoo(${ZOO_TEST_OFM_SUBPATH} ${${use_case}_TEST_OFM})
+        download_file_from_modelzoo(${ZOO_MODEL_VERSION} ${ZOO_TEST_IFM_SUBPATH} ${${use_case}_TEST_IFM})
+        download_file_from_modelzoo(${ZOO_MODEL_VERSION} ${ZOO_TEST_OFM_SUBPATH} ${${use_case}_TEST_OFM})
 
         set(TEST_SRC_GEN_DIR ${CMAKE_BINARY_DIR}/generated/${use_case}/tests/src)
         set(TEST_INC_GEN_DIR ${CMAKE_BINARY_DIR}/generated/${use_case}/tests/include)