search

DingKe / pytorch_workplace

Random miniprojects with pytorch.
MIT License
173 stars 49 forks source link

Focal loss is different from the paper. #1

Open EthanZhangYi opened 7 years ago

EthanZhangYi commented 7 years ago

Hi, https://github.com/DingKe/pytorch_workplace/blob/822b6d2b18bba15cc5f273e901903b7c5bdd94e8/focalloss/loss.py#L33 Here seems should be (1 - exp(log_input)), not (1 - log_input)

DingKe commented 7 years ago

Thanks for pointing out. Have corrected.