Converting to Torch Script

mostafaelaraby / wavegan-pytorch

Pytorch Implementation of wavegan model to generate audio

https://arxiv.org/abs/1802.04208

Apache License 2.0

162 stars 43 forks source link

Closed JordanDavis-1 closed 2 years ago

JordanDavis-1 commented 2 years ago

How would one go about converting the model to torch script?

mostafaelaraby commented 2 years ago

@JordanDavis-1 i recommend checking https://djl.ai/docs/pytorch/how_to_convert_your_model_to_torchscript.htm it will help you convert the saved model to torchscript. I will close the issue now feel free to re-open if you couldn't convert it...