How to make it work with TPU?

ivanvovk / WaveGrad

Implementation of WaveGrad high-fidelity vocoder from Google Brain in PyTorch.

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

402 stars 56 forks source link

How to make it work with TPU? #20

Open mulgray opened 3 years ago

mulgray commented 3 years ago

Thank you for releasing the great project. I'd like to train on TPU(v3) on Google Cloud, how can I change the code to make it work?

ivanvovk commented 3 years ago

@mulgray To be honest, I've never run PyTorch with TPU before, but I know official PyTorch XLA framework may help to connect these things together: https://github.com/pytorch/xla

mulgray commented 3 years ago

Thank you for your reply. I will give it a try.