search

NVIDIA / DeepLearningExamples

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.
13.59k stars 3.24k forks source link

[Tacotron2]Error when convert Tacotron2 to ONNX #898

Open ductho9799 opened 3 years ago

ductho9799 commented 3 years ago

I tried to convert the Tacotron2 Model to ONNX with your code. But I got an error about the LSTM layer in the Encoder. Screenshot from 2021-04-02 17-25-58

NeilFranks commented 3 years ago

I tried to convert the Tacotron2 Model to ONNX with your code. But I got an error about the LSTM layer in the Encoder. Screenshot from 2021-04-02 17-25-58

Ever figure this out? I'm getting the exact same thing.