search

uncbiag / SimpleClick

SimpleClick: Interactive Image Segmentation with Simple Vision Transformers (ICCV 2023)
MIT License
214 stars 33 forks source link

GPU settings #25

Closed che1007 closed 7 months ago

che1007 commented 1 year ago

If I intend to use 2 GPUs for training the branch one code, I just need to adjust the "npus" parameter to 2, correct?

codwest commented 1 year ago

you can set --ngpus=2 or --gpus=0,1 to fullfil your target