UnboundLocalError: local variable 'lr_' referenced before assignmen

mazurowski-lab / finetune-SAM

This is an official repo for fine-tuning SAM to customized medical images.

https://arxiv.org/abs/2404.09957

Apache License 2.0

114 stars 17 forks source link

UnboundLocalError: local variable 'lr_' referenced before assignmen #16

Closed Skwarson96 closed 1 month ago

Skwarson96 commented 1 month ago

Hello! in SingleGPU_train_finetune_noprompt.py when argument ifwarmup is False I get: `UnboundLocalError: local variable 'lr' referenced before assignmentin 136 line [code](https://github.com/mazurowski-lab/finetune-SAM/blob/2e6574e32b8cb7d1437883b03cb2054e0c019118/SingleGPU_train_finetune_noprompt.py#L136). I thing variablelr_` should have a value regardless of the argument if_warmup

Thanks for the great project!

Guhanxue commented 1 month ago

Hi, Thanks for pointing this out. This bug is fixed and please check our updated codes.