Freeze encoder

Joao-L-S-Almeida · Joao-L-S-Almeida · commit 52e15f86b78e · 2024-07-02T11:49:14.000-03:00
Signed-off-by: João Lucas de Sousa Almeida &lt;joao.l.sa.9.3@gmail.com&gt;
diff --git a/tests/manufactured-finetune_prithvi_swin_B.yaml b/tests/manufactured-finetune_prithvi_swin_B.yaml
@@ -127,7 +127,7 @@ model:
     #aux_loss:
     #  aux_head: 0.4
     ignore_index: -1
-    freeze_backbone: false
+    freeze_backbone: true
     freeze_decoder: false
     model_factory: PrithviModelFactory
 
diff --git a/tests/manufactured-finetune_prithvi_swin_L.yaml b/tests/manufactured-finetune_prithvi_swin_L.yaml
@@ -127,7 +127,7 @@ model:
     #aux_loss:
     #  aux_head: 0.4
     ignore_index: -1
-    freeze_backbone: false
+    freeze_backbone: true
     freeze_decoder: false
     model_factory: PrithviModelFactory
 
diff --git a/tests/manufactured-finetune_prithvi_vit_100.yaml b/tests/manufactured-finetune_prithvi_vit_100.yaml
@@ -128,7 +128,7 @@ model:
     #aux_loss:
     #  aux_head: 0.4
     ignore_index: -1
-    freeze_backbone: false
+    freeze_backbone: true
     freeze_decoder: false
     model_factory: PrithviModelFactory
 
diff --git a/tests/manufactured-finetune_prithvi_vit_300.yaml b/tests/manufactured-finetune_prithvi_vit_300.yaml
@@ -128,7 +128,7 @@ model:
     #aux_loss:
     #  aux_head: 0.4
     ignore_index: -1
-    freeze_backbone: false
+    freeze_backbone: true
     freeze_decoder: false
     model_factory: PrithviModelFactory
 
