We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent efbbe60 commit 5559aaaCopy full SHA for 5559aaa
torch2trt/torch2trt.py
@@ -748,7 +748,7 @@ def torch2trt(module,
748
input_names=input_names,
749
output_names=output_names,
750
dynamic_axes={
751
- name: {int(axis): 'axis_%d' % axis for axis in dynamic_axes_flat[index]}
+ name: {int(axis): f'input_{index}_axis_{axis}' for axis in dynamic_axes_flat[index]}
752
for index, name in enumerate(input_names)
753
},
754
opset_version=onnx_opset
0 commit comments