How to run with multiple gpus?

airsplay / R2R-EnvDrop

PyTorch Code of NAACL 2019 paper "Learning to Navigate Unseen Environments: Back Translation with Environmental Dropout"

MIT License

122 stars 25 forks source link

Closed yellowjs0304 closed 4 years ago

yellowjs0304 commented 4 years ago

Hi, Thank u for sharing nice work.

I followed your script, but i can't train this in multiple gpus.

I used "bash run/bt_envdrop.bash 0, 3", but it cannot equally divided into two GPUs.

Could you please tell me how to do it?

airsplay commented 4 years ago

Did not support multiple GPU yet TAT...

yellowjs0304 commented 4 years ago

Thank you for quick response.