From c3699715d6b9c84e3c8ee7240d829f6fb1e82c2e Mon Sep 17 00:00:00 2001
From: Ritu Yadav <40523539+RituYadav92@users.noreply.github.com>
Date: Tue, 17 Sep 2024 07:50:19 +0200
Subject: [PATCH] added regression (segmentors) config

---
 configs/segmentors/reg_upernet.yaml    | 24 ++++++++++++++++++++++++
 configs/segmentors/reg_upernet_mt.yaml |  5 ++---
 2 files changed, 26 insertions(+), 3 deletions(-)
 create mode 100644 configs/segmentors/reg_upernet.yaml

diff --git a/configs/segmentors/reg_upernet.yaml b/configs/segmentors/reg_upernet.yaml
new file mode 100644
index 00000000..c3576b85
--- /dev/null
+++ b/configs/segmentors/reg_upernet.yaml
@@ -0,0 +1,24 @@
+segmentor_name: UPerNet_regress
+task_name: regression
+# time_frames: 2
+#task_model_args:
+  #num_frames: 1
+  #mt_strategy: "ltae" #activated only when if num_frames > 1
+  #num_classes - task parameter passed from the dataset config
+  #wave_list - task parameter passed from the dataset config
+
+channels: 512
+
+loss: 
+  loss_name: MSELoss # WeightedCrossEntropy
+  ignore_index: -1
+
+optimizer:
+  optimizer_name: AdamW
+  lr: 0.0001
+  weight_decay: 0.05
+
+scheduler:
+  scheduler_name: MultiStepLR
+  lr_milestones: [0.6, 0.9]
+
diff --git a/configs/segmentors/reg_upernet_mt.yaml b/configs/segmentors/reg_upernet_mt.yaml
index 6802be33..b3d8faff 100644
--- a/configs/segmentors/reg_upernet_mt.yaml
+++ b/configs/segmentors/reg_upernet_mt.yaml
@@ -1,8 +1,7 @@
-segmentor_name: MTUPerNet
+segmentor_name: MTUPerNet_regress
 task_name: regression
-binary: False
 multi_temporal_strategy: linear
-# time_frames: 2
+time_frames: 12
 #task_model_args:
   #num_frames: 1
   #mt_strategy: "ltae" #activated only when if num_frames > 1