GPU to use
arrrrr3186 opened this issue · 1 comments
arrrrr3186 commented
Hi,
I trying to execute train.py, and scripts is asking for GPU parameter,
as I am using Nvidia, so which GPU i should mention.
JarrentWu1031 commented
Hi,
I trying to execute train.py, and scripts is asking for GPU parameter, as I am using Nvidia, so which GPU i should mention.
Maybe you could just specify --gpu 0 as the default setting.