Skip to content

Commit 7975f7e

Browse files
authored
Merge pull request #3584 from jkamelin/jk/model_suffix
AC: Handle delayed model loading in get_model_file_type
2 parents 8f11193 + 4ffe687 commit 7975f7e

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 hasattr(self, '_model'):
935+
return self._model.suffix
936+
else:
937+
return None
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 hasattr(self, '_model'):
971+
return self._model.suffix
972+
else:
973+
return None
971974

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

0 commit comments

Comments
 (0)