Skip to content

Commit c166973

Browse files
authored
Merge pull request #3632 from eaidova/ea/mo_deprecated
OMZ comverter: remove deprecated mo parameters usage
2 parents 214d362 + 77251a9 commit c166973

File tree

3 files changed

+2
-3
lines changed

3 files changed

+2
-3
lines changed

models/public/mozilla-deepspeech-0.6.1/model.yml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,5 @@ model_optimizer_args:
4545
- --input_model=$dl_dir/deepspeech-0.6.1-models/output_graph.pb
4646
- --freeze_placeholder_with_value=input_lengths->[16]
4747
- --output=logits,cudnn_lstm/rnn/multi_rnn_cell/cell_0/cudnn_compatible_lstm_cell/GatherNd,cudnn_lstm/rnn/multi_rnn_cell/cell_0/cudnn_compatible_lstm_cell/GatherNd_1
48-
- --disable_nhwc_to_nchw
4948
framework: tf
5049
license: https://raw.githubusercontent.com/mozilla/DeepSpeech/master/LICENSE

models/public/mozilla-deepspeech-0.8.2/model.yml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,5 @@ model_optimizer_args:
4646
- --input_model=$conv_dir/deepspeech-0.8.2-models.pb
4747
- --freeze_placeholder_with_value=input_lengths->[16]
4848
- --output=logits,cudnn_lstm/rnn/multi_rnn_cell/cell_0/cudnn_compatible_lstm_cell/GatherNd,cudnn_lstm/rnn/multi_rnn_cell/cell_0/cudnn_compatible_lstm_cell/GatherNd_1
49-
- --disable_nhwc_to_nchw
5049
framework: tf
5150
license: https://raw.githubusercontent.com/mozilla/DeepSpeech/master/LICENSE

tools/model_tools/src/openvino/model_zoo/omz_converter.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,10 +125,11 @@ def convert(reporter, model, output_dir, args, mo_props, requested_precisions):
125125
expanded_mo_args.append('--layout={}'.format(layout_string))
126126
if shape_string:
127127
expanded_mo_args.append('--input_shape={}'.format(shape_string))
128+
if data_type == "FP16":
129+
expanded_mo_args.append("--compress_to_fp16")
128130

129131
mo_cmd = [*mo_props.cmd_prefix,
130132
'--framework={}'.format(model_format),
131-
'--data_type={}'.format(data_type),
132133
'--output_dir={}'.format(output_dir / model.subdirectory / model_precision),
133134
'--model_name={}'.format(model.name),
134135
'--input={}'.format(','.join(input.name for input in model.input_info)),

0 commit comments

Comments
 (0)