This repository has been archived by the owner on May 28, 2021. It is now read-only.

About the loss computation #53

Open

chrisway613 opened this issue Apr 11, 2020 · 1 comment

chrisway613 commented Apr 11, 2020

I was confused about why use BCE to compute the loss of 'x' & 'y', could it be MSE?

YOLOv3_PyTorch/nets/yolo_loss.py

Line 55 in c6b4837

loss_x = self.bce_loss(x * mask, tx * mask)

BigDataHa commented Jul 16, 2020

I was confused about why use BCE to compute the loss of 'x' & 'y', could it be MSE?

YOLOv3_PyTorch/nets/yolo_loss.py

Line 55 in c6b4837

loss_x = self.bce_loss(x * mask, tx * mask)

I have same question with you

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