search

lufficc / SSD

High quality, fast, modular reference implementation of SSD in PyTorch
MIT License
1.52k stars 384 forks source link

Which loss function is used? #143

Closed cyberclone12 closed 4 years ago

cyberclone12 commented 4 years ago

@lufficc

lufficc commented 4 years ago

https://github.com/lufficc/SSD/blob/master/ssd/modeling/box_head/loss.py#L34-L39

cross_entropy loss and smooth_l1_loss