WaveGrad icon indicating copy to clipboard operation
WaveGrad copied to clipboard

Published 20 hours ago • verified publisher icon ivanvovk

How to make it work with TPU?

Open mulgray opened this issue 3 years ago • 2 comments

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?

mulgray avatar Mar 25 '21 09:03 mulgray

@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

ivanvovk avatar Mar 26 '21 08:03 ivanvovk

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

mulgray avatar Mar 27 '21 03:03 mulgray