Update error msg

pycroscopy · Apr 12, 2023 · f8266bc · f8266bc
1 parent 412e0d8
commit f8266bc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/atomai/losses_metrics/losses.py b/atomai/losses_metrics/losses.py
@@ -167,7 +167,8 @@ def select_loss(loss: str, nb_classes: int = None, **kwargs):
     else:
         raise NotImplementedError(
             "Select Dice loss ('dice'), focal loss ('focal') "
-            " cross-entropy loss ('ce') or means-squared error ('mse')"
+            " cross-entropy loss ('ce'), means-squared error ('mse'),"
+            " multitask loss (multitask_nll and multitask_ce)"
             " or pass your custom loss function"
         )
     return criterion