File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed
tools/accuracy_checker/openvino/tools/accuracy_checker/evaluators/custom_evaluators Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -195,13 +195,14 @@ def automatic_model_search(self, network_info):
195
195
return model , weights
196
196
197
197
def set_input_and_output (self ):
198
- has_info = hasattr (self .exec_network , 'input_info' )
199
- input_info = self .exec_network .input_info if has_info else self .exec_network .inputs
198
+ network = self .exec_network if self .exec_network is not None else self .network
199
+ has_info = hasattr (network , 'input_info' )
200
+ input_info = network .input_info if has_info else network .inputs
200
201
input_blob = next (iter (input_info ))
201
202
with_prefix = input_blob .startswith (self .default_model_suffix )
202
203
if self .input_blob is None or with_prefix != self .with_prefix :
203
204
if self .output_blob is None :
204
- output_blob = next (iter (self . exec_network .outputs ))
205
+ output_blob = next (iter (network .outputs ))
205
206
else :
206
207
output_blob = (
207
208
'_' .join ([self .default_model_suffix , self .output_blob ])
You can’t perform that action at this time.
0 commit comments