kevinzakka / pytorch-goodies

PyTorch Boilerplate For Research
603 stars 71 forks source link

Tversky loss with ignore_label #13

Open hinriksnaer opened 3 years ago

hinriksnaer commented 3 years ago

Is there a simple way to incorporate ignore_label into the tversky_loss function?