how to load pytorch model into cpp

NVIDIA / tacotron2

Tacotron 2 - PyTorch implementation with faster-than-realtime inference

BSD 3-Clause "New" or "Revised" License

5.08k stars 1.38k forks source link

how to load pytorch model into cpp #198

Open zhanglq95 opened 5 years ago

zhanglq95 commented 5 years ago

I want to use this TTS model in cpp, but after I reading the example of pytorch documention, i'm confused of turning the decoder module to torch.jit.script. Is there some suggestion about how to deploy the pytorch model in cpp? thanks very much!

rafaelvalle commented 5 years ago

The model is exportable using torch.jit.script. You will need to make some changes to comply with the requirements of torch.jit. Let us know if you hit any blocks in the process of converting the model to torch.jit.script.