added unet decoder config for regression (#121)

yurujaja · Nov 19, 2024 · 5cc4b34 · 5cc4b34
1 parent 9a795ad
commit 5cc4b34
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/configs/decoder/reg_unet.yaml b/configs/decoder/reg_unet.yaml
@@ -0,0 +1,5 @@
+_target_: pangaea.decoders.unet.UNet
+
+encoder: null
+num_classes: 1
+finetune: True # ${finetune}