search

mlyg / unified-focal-loss

Apache License 2.0
150 stars 22 forks source link

Is there a pytorch version #2

Closed Songxinlei closed 2 years ago

Songxinlei commented 3 years ago

Is there a pytorch version？

mlyg commented 3 years ago

I am afraid not at the moment. However, I will be working on translating these loss functions into pytorch soon so I will update the repository once that is ready.

In a previous issue opened by another user, it appears they have converted the 'Focal Tversky loss' into pytorch - feel free to take a look and see whether it is something you can use in the meantime.