minor spacing fix

NRCan · Feb 6, 2025 · a125b0f · a125b0f
1 parent 49e8929
commit a125b0f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/geo_deep_learning/tasks_with_models/segmentation_unetplus.py b/geo_deep_learning/tasks_with_models/segmentation_unetplus.py
@@ -50,8 +50,7 @@ def __init__(self,
         if weights_from_checkpoint_path:
             print(f"Loading weights from checkpoint: {weights_from_checkpoint_path}")
             checkpoint = torch.load(weights_from_checkpoint_path)
-            self.load_state_dict(checkpoint['state_dict'])    
-
+            self.load_state_dict(checkpoint['state_dict'])
         self.loss = loss
         num_classes = num_classes + 1 if num_classes == 1 else num_classes
         self.iou_metric = MeanIoU(num_classes=num_classes,