tf-gqn icon indicating copy to clipboard operation
tf-gqn copied to clipboard

Published 20 hours ago • verified publisher icon ogroth

How to train using multi-GPU

Open phongnhhn92 opened this issue 4 years ago • 1 comments

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.

phongnhhn92 avatar Sep 30 '19 17:09 phongnhhn92

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.

ogroth avatar Sep 30 '19 17:09 ogroth