search

tianrun-chen / SAM-Adapter-PyTorch

Adapting Meta AI's Segment Anything to Downstream Tasks with Adapters and Prompts
MIT License
850 stars 75 forks source link

How to run on one GPU? #49

Open CatByteDance opened 10 months ago

CatByteDance commented 10 months ago

Hi, thanks for your excellent work. I am new to this area, I am wondering if I can use only one GPU to train this model, if posssible, how should I change the setting, thanks.