How can I use multiple GPU in train? please...

tyliupku / wiki2bio

Code for AAAI2018 paper "Table-to-text Generation by Structure-aware Seq2seq Learning"

https://arxiv.org/abs/1711.09724

152 stars 46 forks source link

Open dongin1009 opened 3 years ago

dongin1009 commented 3 years ago

It maybe related to tf.device .. But I can't find right method to use multiple GPU in TF1 for this code. Help me..