Skip to content

Commit 0a6277a

Browse files
authored
Switch from False to None base_models.py
1 parent 7c60a41 commit 0a6277a

File tree

1 file changed

+2
-2
lines changed
  • tools/accuracy_checker/accuracy_checker/evaluators/custom_evaluators

1 file changed

+2
-2
lines changed

tools/accuracy_checker/accuracy_checker/evaluators/custom_evaluators/base_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def __init__(self, network_info, launcher, suffix=None, delayed_model_loading=Fa
137137
self.input_blob = None
138138
self.with_prefix = False
139139
self.is_dynamic = False
140-
self._dump_first_infer_data = network_info.get('_dump_first_infer_data', False)
140+
self._dump_first_infer_data = network_info.get('_dump_first_infer_data', None)
141141
if not delayed_model_loading:
142142
self.load_model(network_info, launcher, log=True)
143143

@@ -412,7 +412,7 @@ def infer(self, input_data, raw_results=False):
412412
print_info(f'Storing first inference data to {self._dump_first_infer_data}')
413413
dump_inf_data = {'input': feed_dict, 'output': res_outputs}
414414
pickle.dump(dump_inf_data, file)
415-
self._dump_first_infer_data = False
415+
self._dump_first_infer_data = None
416416
if raw_results:
417417
return res_outputs, outputs
418418
return res_outputs

0 commit comments

Comments
 (0)