Pass input_names and output_names directly in TorchOnnxExportJob

rwth-i6 / i6_core

Sisyphus recipies for ASR

Mozilla Public License 2.0

16 stars 23 forks source link

Pass input_names and output_names directly in TorchOnnxExportJob #510

Closed kuacakuaca closed 5 months ago

kuacakuaca commented 5 months ago

https://github.com/rwth-i6/returnn/issues/1517

albertz commented 5 months ago

Needs some black reformatting now. Also not sure if the docstring is maybe too long. Then needs some line break there as well.

But otherwise looks good to me.

NeoLegends commented 3 months ago

After this PR I'm having some issues w/ dynamic axes (e.g. data:size1) not being specified by the job, but being inferred by RETURNN. I think these should be added to the list of inferred input and output names, too.