New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

why the train loss without iou_prediction? #124

Open

cvhuang opened this issue Jul 10, 2024 · 0 comments

cvhuang commented Jul 10, 2024 •

edited

Loading

efficientvit/efficientvit/samcore/trainer/sam_trainer.py

Line 190 in 5375cf9

loss_i = loss_mask_i * 20 + loss_dice_i

I think you may forget iou_prediction part. Otherwise, how the parameter of iou_prediction head update.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment