train_learnable_sam.py : TypeError: to() received an invalid combination of arguments - got (detype=torch.dtype, )

Qsingle / LearnablePromptSAM

Try to use the SAM-ViT as the backbone to create the learnable prompt for semantic segmentation

Apache License 2.0

77 stars 13 forks source link

Open krishnaadithya opened 1 year ago

krishnaadithya commented 1 year ago

line 176 train_learnable_sam.py "detype" needs to be changed to dtype

Qsingle commented 1 year ago

Thank you for your advice, we will fix it in next version.