SysCV / sam-hq

Segment Anything in High Quality [NeurIPS 2023]
https://arxiv.org/abs/2306.01567
Apache License 2.0
3.73k stars 224 forks source link

Hello author, how do you measure loss_mask and loss dice, do you have any references? #61

Open moliniao opened 1 year ago

moliniao commented 1 year ago

Custom data set (10k), using the author's method, after a round of training, loss_dice and loss_mask have been fluctuating at a certain value, how do you know that it will converge if you continue to train?

image
lkeab commented 1 year ago

Hi, you can draw a loss curve and observe the tendency of loss changing.

moliniao commented 1 year ago

Hi, you can draw a loss curve and observe the tendency of loss changing.

Can you provide some code samples?