added freeze_encoder param

geo_deep_learning/tasks_with_models/segmentation_segformer.py

@@ -25,6 +25,7 @@ def __init__(self,
                  std: List[float],
                  data_type_max: float,
                  loss: Callable,
+                 freeze_encoder: bool = False,
                  weights: str = None,
                  class_labels: List[str] = None,
                  class_colors: List[str] = None,
@@ -38,7 +39,7 @@ def __init__(self,
         self.data_type_max = data_type_max
         self.class_colors = class_colors
         self.num_classes = num_classes
-        self.model = SegFormer(encoder, in_channels, weights, self.num_classes)
+        self.model = SegFormer(encoder, in_channels, weights, freeze_encoder, self.num_classes)
         if weights_from_checkpoint_path:
             print(f"Loading weights from checkpoint: {weights_from_checkpoint_path}")
             checkpoint = torch.load(weights_from_checkpoint_path)