I intend to use torch.onnx.export to convert the DeepFloyd model to onnx format, documentation on this function is https://glaringlee.github.io/onnx.html#functions.
The problem I'm having is that I don't know what args, _inputnames, _outputnames and _dynamicaxes arguments should be input when using this function.
in other words, I don't know what the inputs, input_names, output_names and dynamic_axes of the DeepFloyd model are. Where can I get information about these parameters?
I intend to use torch.onnx.export to convert the DeepFloyd model to onnx format, documentation on this function is https://glaringlee.github.io/onnx.html#functions. The problem I'm having is that I don't know what args, _inputnames, _outputnames and _dynamicaxes arguments should be input when using this function. in other words, I don't know what the inputs, input_names, output_names and dynamic_axes of the DeepFloyd model are. Where can I get information about these parameters?