How to train the network in three GPUs

[XGBoost]

I have noticed you have mentioned that it takes four hours to finish the training on three NVIDIA GTX 1080 Ti GPUs. However, you do not describe how to train the network on three GPUs in README.md.
When I run
python train.py --id resnet50_rnn --use_rnn
, it only takes a single GPU, and the batch size of it is eight which is different from that mentioned in your paper.
Could you please describe the process of training in detail.

[sunset1995]

Hi @XGBoost
To train on multiple GPU, you have to modify few lines of code.
You can check https://pytorch.org/tutorials/beginner/former_torchies/parallelism_tutorial.html
This repo only work on single GPU.