taki0112 / SPADE-Tensorflow

Simple Tensorflow implementation of "Semantic Image Synthesis with Spatially-Adaptive Normalization" a.k.a. GauGAN, SPADE (CVPR 2019 Oral)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Multi-GPU training

HanYangZhao opened this issue · comments

Hi,

Thanks for all the hard work. I was wondering how one would add support for multi-gpu training. I see that in SPADE.py we do a pre-fetch to the GPU in the build_model(), but I'm somewhat stumped on how multi-gpu would work

Thanks