How to train using multi-GPU

[phongnhhn92]

Thanks for the code!
I wonder is there any way to train this model using 2 GPU. Right now I am training GQN with shepard_metzler_5_parts dataset and only 1 GPU is training.

[ogroth]

Hi @phongnhhn92 , I haven't really looked into multi-GPU training, yet. But I think, you'd be able to specify the parameters for it in the GPU options of the estimator.