BloodAxe / pytorch-toolbelt

PyTorch extensions for fast R&D prototyping and Kaggle farming
MIT License
1.51k stars 119 forks source link

fix bug in lovasz loss #62

Closed seefun closed 3 years ago

seefun commented 3 years ago

https://github.com/BloodAxe/pytorch-toolbelt/issues/61

BloodAxe commented 3 years ago

Thanks for your PR, indeed this looks to be a correct implementation.