search

buxiangzhiren / DDCap

MIT License
83 stars 11 forks source link

How to use one GPU for training. #18

Open shenyiyangsyy opened 1 year ago

shenyiyangsyy commented 1 year ago

Hello, how can I use only one GPU for training this model?

buxiangzhiren commented 1 year ago

You just need to change "--nproc_per_node" from 8 to 1 when you run the train.py.