Enable encoder freezing in SegFormer configuration

NRCan · Dec 9, 2024 · eafca1e · eafca1e
1 parent 3f0f2de
commit eafca1e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/configs/segformer_config_RGB.yaml b/configs/segformer_config_RGB.yaml
@@ -57,6 +57,7 @@ model:
       class_path: segmentation_models_pytorch.losses.DiceLoss
       init_args:
         mode: "multiclass"
+    freeze_encoder: true
     class_labels: ["background", "fore", "hydro", "roads", "buildings"]
     class_colors: ["#000000", "#008000", "#0000FF", "#FFFF00", "#FF0000"]
     weights_from_checkpoint_path: null