Skip to content

Commit bb96615

Browse files
committed
quartznet update onnx export command for preserving dynamic shapes
1 parent 2f1b1c0 commit bb96615

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

models/public/quartznet-15x5-en/model.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,7 @@ conversion_to_onnx_args:
101101
- --model-param=decoder_weights=r"$dl_dir/models/.nemo_tmp/JasperDecoderForCTC.pt"
102102
- --input-names=audio_signal
103103
- --output-names=output
104+
- '--conversion-param=dynamic_axes={"audio_signal": {0: "batch_size, 2: "wave_len"}, {"output: {0: "batch_size", 2: "wave_len"}}'
104105
input_info:
105106
- name: audio_signal
106107
shape: [1, 64, 128]

0 commit comments

Comments
 (0)