bruh

microsoft · Feb 25, 2024 · 3ea459e · 3ea459e
1 parent c204504
commit 3ea459e
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/torchgeo/trainers/segmentation.py b/torchgeo/trainers/segmentation.py
@@ -110,9 +110,7 @@ def configure_losses(self) -> None:
                 i for i in range(self.hparams["num_classes"]) if i != ignore_index
             ]
 
-            self.criterion = smp.losses.JaccardLoss(
-                mode="multiclass", classes=classes
-            )
+            self.criterion = smp.losses.JaccardLoss(mode="multiclass", classes=classes)
         elif loss == "focal":
             self.criterion = smp.losses.FocalLoss(
                 "multiclass", ignore_index=ignore_index, normalized=True