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

How to train on multi GPU? #39

Closed creke closed 5 years ago

creke commented 5 years ago

Hi , I'm new to tensorflow, and I trained the model on a multi GPU server. But it took only ONE gpu. I want to make all of my gpus useful, is there a way to do so? Thank you.

kicGit commented 5 years ago

me too

tafseerahmed commented 5 years ago

I am working on a PyTorch multi GPU-training version using Pytorch-lightning

taki0112 commented 5 years ago

We didn't implement it in multi-gpu. I think you'll need a little code change to do that.

zhangyunming commented 5 years ago

how to change the code if we use multi-gpus it only can train light version on 11G memoy card