Multi-GPU training

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