From 3f0f2de94617266775bc2a55865d7dabc4144224 Mon Sep 17 00:00:00 2001
From: valhassan <victor.alhassan@canada.ca>
Date: Mon, 9 Dec 2024 14:21:41 -0500
Subject: [PATCH] Enable encoder freezing in DOFA configuration

---
 configs/dofa_config_RGB.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/configs/dofa_config_RGB.yaml b/configs/dofa_config_RGB.yaml
index 5e4f7cad..9c2e61fb 100644
--- 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