sam-hq icon indicating copy to clipboard operation
sam-hq copied to clipboard
verified publisher icon SysCV

Can the train.py ssupport the training for Light HQ-SAM?

Open rbli-john opened this issue 2 years ago • 1 comments

Hi,

Currently in the train.py, only model_type in ["vit_b","vit_l", "vit_h"] are supported. My question is, can I use the train.py to train the Light HQ-SAM (model_type = "vit_tiny")? If yes, how can I modify train.py to support the light version?

Thanks.

rbli-john avatar Oct 26 '23 22:10 rbli-john

Hello, I have the same problem, have you solved it yet?

bobo59 avatar Sep 24 '24 07:09 bobo59