Train on multiple GPU

#6
by vladciocan88 - opened

Hi, im trying to train using the colab script but with 2 A6000, but it only runs on one. Im kinda new to this and will appreciate any help.

Sorry for the inconvenience. I have added a comment in the code stating how to make it run in more than 1 GPU

Sign up or log in to comment