ZrrSkywalker / Personalize-SAM

Personalize Segment Anything Model (SAM) with 1 shot in 10 seconds
MIT License
1.5k stars 101 forks source link

persam_f.py: error: argument --lr: invalid int value: '0.01' #19

Closed xyqlx closed 1 year ago

xyqlx commented 1 year ago

https://github.com/ZrrSkywalker/Personalize-SAM/blob/c1effd8d8b2dc166dcf8e6927602123dd0387c96/persam_f.py#L27

should be changed to

parser.add_argument('--lr', type=float, default=1e-3) 
ZrrSkywalker commented 1 year ago

Thanks very much for pointing out! We have fixed this bug.