Skip to content

Commit bfc67ae

Browse files
authored
demos: fix speech_recognition_deepspeech (#3949)
It seems MO added :0 to the names for mozilla-deepspeech-0.6.1 and mozilla-deepspeech-0.8.2 depite the fact that the config doesn't have :0: https://github.com/openvinotoolkit/open_model_zoo/blob/828890a839480240467bffc0c4165e37b38719a8/models/public/mozilla-deepspeech-0.6.1/model.yml#L47
1 parent 828890a commit bfc67ae

File tree

1 file changed

+3
-3
lines changed
  • demos/speech_recognition_deepspeech_demo/python/asr_utils

1 file changed

+3
-3
lines changed

demos/speech_recognition_deepspeech_demo/python/asr_utils/profiles.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,10 +26,10 @@
2626
# in_state_c, in_state_h, out_state_c, out_state_h, in_data, out_data (str), IR node names
2727
'in_state_c': 'previous_state_c:0',
2828
'in_state_h': 'previous_state_h:0',
29-
'out_state_c': 'cudnn_lstm/rnn/multi_rnn_cell/cell_0/cudnn_compatible_lstm_cell/GatherNd',
30-
'out_state_h': 'cudnn_lstm/rnn/multi_rnn_cell/cell_0/cudnn_compatible_lstm_cell/GatherNd_1',
29+
'out_state_c': 'cudnn_lstm/rnn/multi_rnn_cell/cell_0/cudnn_compatible_lstm_cell/GatherNd:0',
30+
'out_state_h': 'cudnn_lstm/rnn/multi_rnn_cell/cell_0/cudnn_compatible_lstm_cell/GatherNd_1:0',
3131
'in_data': 'input_node',
32-
'out_data': 'logits', # Despite being named logits, output is probabilities after softmax
32+
'out_data': 'logits:0', # Despite being named logits, output is probabilities after softmax
3333

3434
# === CTC decoder and LM parameters ===
3535
# log_probs (bool), True is input data contains base e log(probabilities), False if simply probabilities.

0 commit comments

Comments
 (0)