Мой код
a=torch.randn(1,80,100,requires_grad=True)
torch.onnx.export(waveglow,a, "waveglow.onnx")
Я пытаюсь экспортировать модель PyTorch в формат ONNX, чтобы я мог использовать ее в TensorRT.во время тестирования моей модели в PyTorch размер входного тензора равен (1,80, x), где x изменяется в зависимости от длины входного текста (используемая модель - модель TTS с именем waveglow)
при попытке запуститьПриведенный выше код для экспорта модели в onnx, я всегда получаю эту ошибку
RuntimeError: Only tuples, lists and Variables supported as JIT inputs, but got NoneType
Пожалуйста, помогите