Skip to content

Commit 8f5b880

Browse files
authored
AC: fix lpcnet quantization (#3175)
1 parent 323be85 commit 8f5b880

File tree

1 file changed

+18
-1
lines changed
  • tools/accuracy_checker/openvino/tools/accuracy_checker/evaluators/custom_evaluators

1 file changed

+18
-1
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/evaluators/custom_evaluators/lpcnet_evaluator.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,8 @@ def __init__(self, network_info, launcher, models_args, adapter_info, is_blob=No
6363
self.decoder = create_model(network_info['decoder'], launcher, self._decoder_mapping, 'decoder',
6464
delayed_model_loading)
6565
self.adapter = create_adapter(adapter_info)
66-
self.update_inputs_outputs_info()
66+
if not delayed_model_loading:
67+
self.update_inputs_outputs_info()
6768
self.adapter.output_blob = 'audio'
6869

6970
self.with_prefix = False
@@ -123,6 +124,10 @@ def update_inputs_outputs_info(self, with_prefix):
123124
self.periods_input = generate_layer_name(self.periods_input, self.default_model_suffix+'_', with_prefix)
124125
if hasattr(self, 'outputs'):
125126
self.output = postprocess_output_name(self.output, self.outputs, raise_error=False)
127+
if self.output not in self.outputs:
128+
self.output = postprocess_output_name(
129+
generate_layer_name(self.output, self.default_model_suffix+'_', with_prefix),
130+
self.outputs, raise_error=False)
126131

127132

128133
class EncoderDLSDKModel(EncoderModel, TTSDLSDKModel):
@@ -240,8 +245,20 @@ def update_inputs_outputs_info(self, with_prefix):
240245
self.rnn_input2 = generate_layer_name(self.rnn_input2, prefix, with_prefix)
241246
if hasattr(self, 'outputs'):
242247
self.output = postprocess_output_name(self.output, self.outputs, raise_error=False)
248+
if self.output not in self.outputs:
249+
self.output = postprocess_output_name(
250+
generate_layer_name(self.output, self.default_model_suffix + '_', with_prefix),
251+
self.outputs, raise_error=False)
243252
self.rnn_output1 = postprocess_output_name(self.rnn_output1, self.outputs, raise_error=False)
253+
if self.rnn_output1 not in self.outputs:
254+
self.rnn_output1 = postprocess_output_name(
255+
generate_layer_name(self.rnn_output1, self.default_model_suffix + '_', with_prefix),
256+
self.outputs, raise_error=False)
244257
self.rnn_output2 = postprocess_output_name(self.rnn_output2, self.outputs, raise_error=False)
258+
if self.rnn_output2 not in self.outputs:
259+
self.rnn_output2 = postprocess_output_name(
260+
generate_layer_name(self.rnn_output2, self.default_model_suffix + '_', with_prefix),
261+
self.outputs, raise_error=False)
245262

246263

247264
class DecoderONNXModel(BaseONNXModel, DecoderModel):

0 commit comments

Comments
 (0)