Skip to content

Commit 534a364

Browse files
committed
handle delayed model loading in get_model_file_type
1 parent 8f11193 commit 534a364

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/launcher/dlsdk_launcher.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -931,7 +931,10 @@ def _set_preprocess(self, preprocess):
931931
self.disable_resize_to_input = preprocess.ie_processor.has_resize()
932932

933933
def get_model_file_type(self):
934-
return self._model.suffix
934+
if self._delayed_model_loading:
935+
return None
936+
else:
937+
return self._model.suffix
935938

936939
def input_shape(self, input_name):
937940
if input_name in self._partial_shapes:

tools/accuracy_checker/openvino/tools/accuracy_checker/launcher/openvino_launcher.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -967,7 +967,10 @@ def _set_preprocess(self, preprocess):
967967
self.disable_resize_to_input = preprocess.ie_processor.has_resize()
968968

969969
def get_model_file_type(self):
970-
return self._model.suffix
970+
if self._delayed_model_loading:
971+
return None
972+
else:
973+
return self._model.suffix
971974

972975
def get_infer_queue(self, log=True):
973976
if self.config.get('num_requests', 'AUTO') == 'AUTO':

0 commit comments

Comments
 (0)