ZebraPose icon indicating copy to clipboard operation
ZebraPose copied to clipboard

Does loss curves look right?

Open yinguoxiangyi opened this issue 1 year ago • 4 comments

The following is the loss curve of my own dataset. Does it look right? it seems that binary code loss didn't decrease. loss

yinguoxiangyi avatar Oct 11 '23 06:10 yinguoxiangyi

The binary code loss is only calculated for pixels within the predicted mask.

For debugging, you can set mask_binary_code_loss to False to check if the binary code loss converges.

suyz526 avatar Oct 11 '23 13:10 suyz526

According to your past experiments, ideally, which value the binary_code_loss will decrease to ?

yinguoxiangyi avatar Oct 12 '23 05:10 yinguoxiangyi

I don't have any logs available at this moment. Is your object symmetrical?

suyz526 avatar Oct 13 '23 09:10 suyz526

Hi @yinguoxiangyi ,

I observed the same behaviour when training Zebrapose on my custom dataset:

mask_binary_code_loss = False results in a nice exponentially decreasing curve: mask_binary_code_loss_False

mask_binary_code_loss = True results in a somewhat constant graph at first: mask_binary_code_loss_True

I continued the training for a bit longer (with mask_binary_code_loss = True), and at some point the binary code loss starts to decrease. Though note the different scale of the y axis compared to the previous screenshot: Screenshot from 2023-11-02 09-27-52

The test-time metrics are increasing, so it seems like the training is running just fine, despite the unusual looking graph.

jonashein avatar Nov 02 '23 08:11 jonashein