search

taki0112 / UGATIT

Official Tensorflow implementation of U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translation (ICLR 2020)
MIT License
6.17k stars 1.04k forks source link

Do you have a TPU version? #31

Closed thewaifuai closed 5 years ago

thewaifuai commented 5 years ago

I am actively working on building a TPU version of UGATIT. By any chance do you already have or plan to work on a TPU implementation? The benefit is that (hopefully) instead of taking days it would take hours to train.

sebasfavaron commented 5 years ago

Dont have a TPU, but looking forward to people using this and sharing pretrained weights, lol. Keep at it!

tafseerahmed commented 5 years ago

I am actively working on building a TPU version of UGATIT. By any chance do you already have or plan to work on a TPU implementation? The benefit is that (hopefully) instead of taking days it would take hours to train.

How about use PyTorch Lightning to make a multi GPU training Version.