-
Notifications
You must be signed in to change notification settings - Fork 17
Open
Description
Hi,
I am trying to use multiple GPUs on my workstation for your code. I thus use GID=0,1,2,3 in the command to start a training session. However, it seems that it's still using only 1 GPU.
Going through your code, I was unable to find DataParallel anywhere in the code. I am wondering whether if your code originally supports multi-GPU training.
If not, I might be able to take a look at.
Metadata
Metadata
Assignees
Labels
No labels