Skip to content

Commit 1ab3d1f

Browse files
committed
edit exception message to include dice loss
Signed-off-by: Carlos Gomes <[email protected]>
1 parent 2842f2d commit 1ab3d1f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/terratorch/tasks/segmentation_tasks.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -158,7 +158,7 @@ def configure_losses(self) -> None:
158158
elif loss == "dice":
159159
self.criterion = smp.losses.DiceLoss("multiclass", ignore_index=ignore_index)
160160
else:
161-
exception_message = f"Loss type '{loss}' is not valid. Currently, supports 'ce', 'jaccard' or 'focal' loss."
161+
exception_message = f"Loss type '{loss}' is not valid. Currently, supports 'ce', 'jaccard', 'dice' or 'focal' loss."
162162
raise ValueError(exception_message)
163163

164164
def configure_metrics(self) -> None:

0 commit comments

Comments
 (0)