multi GPU

[vijaye12]

Hi team,

Thanks for the nice paper and opensourcing the code.

I am trying to run the code in multi gpu fashion using 8 gpus, but I get the below error RuntimeError: grad can be implicitly created only for scalar outputs

Does the current code support Multi-GPU ?