Only one gpu

Open wangfangfangcn opened this issue 2 years ago • 2 comments

If I want to modify it to single card training, which part of the code will need to be modified? How to modify?

Mar 14 '23 08:03 wangfangfangcn

Me too. I don't know how to modified it.

Feb 18 '24 12:02 chenshao107

in scripts/train.sh --nproc_per_node=Your number of GPU

Feb 18 '24 13:02 chenshao107