Upload config.yaml (#14)

- Upload config.yaml (f4d2b3111e48b5c06f715cedca3d49eaeaa28910)


Co-authored-by: Helen Tamura-Wicks <[email protected]>

config.yaml

@@ -1,4 +1,4 @@
-# lightning.pytorch==2.1.1
+custom_modules_path: ./../granite_geo_flood/custom_modules # path relative to dir from which terratorch is being called
 seed_everything: 0
 trainer:
   accelerator: gpu # we can also use auto or cpu
@@ -8,6 +8,7 @@ trainer:
   logger: True # will use tensorboardlogger
 
   callbacks:
+    - class_path: RichProgressBar
     - class_path: LearningRateMonitor
       init_args:
         logging_interval: epoch
@@ -60,8 +61,8 @@ data:
     train_split: ./../data/regions/combined_uki_spain/splits/flood_train_data.txt
     test_split: ./../data/regions/combined_uki_spain/splits/flood_test_data.txt
     val_split: ./../data/regions/combined_uki_spain/splits/flood_val_data.txt
-    img_grep: "*_image.tif"
-    label_grep: "*_label.tif"
+    img_grep: "*.tif"
+    label_grep: "*.tif"
     no_label_replace: -1
     no_data_replace: 0
     means:
@@ -95,8 +96,7 @@ model:
       decoder: FCNDecoder
       backbone_pretrained: true
       backbone: granite_geospatial_uki
-      backbone_pretrained_cfg_overlay:
-        file: ./../data/checkpoints/granite_geospatial_uki.pt
+      backbone_ckpt_path: ./../data/checkpoints/granite_geospatial_uki.pt
       backbone_pretrain_img_size: 512
       decoder_channels: 256
       # in_channels: 9