Skip to content

Commit

Permalink
edit exception message to include dice loss
Browse files Browse the repository at this point in the history
Signed-off-by: Carlos Gomes <[email protected]>
  • Loading branch information
CarlosGomes98 committed May 30, 2024
1 parent 2842f2d commit 1ab3d1f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/terratorch/tasks/segmentation_tasks.py
Original file line number Diff line number Diff line change
Expand Up @@ -158,7 +158,7 @@ def configure_losses(self) -> None:
elif loss == "dice":
self.criterion = smp.losses.DiceLoss("multiclass", ignore_index=ignore_index)
else:
exception_message = f"Loss type '{loss}' is not valid. Currently, supports 'ce', 'jaccard' or 'focal' loss."
exception_message = f"Loss type '{loss}' is not valid. Currently, supports 'ce', 'jaccard', 'dice' or 'focal' loss."
raise ValueError(exception_message)

def configure_metrics(self) -> None:
Expand Down

0 comments on commit 1ab3d1f

Please sign in to comment.