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