From 0a71fd5d2df623e0c1df11691efcdaf54e17a43f Mon Sep 17 00:00:00 2001
From: Ritu Yadav <40523539+RituYadav92@users.noreply.github.com>
Date: Mon, 18 Nov 2024 22:11:07 +0100
Subject: [PATCH] added unet decoder config for regression

---
 configs/decoder/reg_unet.yaml | 5 +++++
 1 file changed, 5 insertions(+)
 create mode 100644 configs/decoder/reg_unet.yaml

diff --git a/configs/decoder/reg_unet.yaml b/configs/decoder/reg_unet.yaml
new file mode 100644
index 0000000..e9c0940
--- /dev/null
+++ b/configs/decoder/reg_unet.yaml
@@ -0,0 +1,5 @@
+_target_: pangaea.decoders.unet.UNet
+
+encoder: null
+num_classes: 1
+finetune: True # ${finetune}
\ No newline at end of file