Enable encoder freezing in DOFA configuration

NRCan · Dec 9, 2024 · 3f0f2de · 3f0f2de
1 parent b4c1b43
commit 3f0f2de
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/configs/dofa_config_RGB.yaml b/configs/dofa_config_RGB.yaml
@@ -58,6 +58,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