Upload config.yaml

config.yaml

@@ -1,7 +1,7 @@
-# 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: cpu
+  accelerator: auto
   strategy: auto
   devices: auto
   num_nodes: 1
@@ -93,8 +93,9 @@ model:
   init_args:
     model_args:
       decoder: FCNDecoder
-      backbone_pretrained: false
-      backbone: granite_geospatial_uki      
+      backbone_pretrained: true
+      backbone: granite_geospatial_uki
+      backbone_ckpt_path: ../data/checkpoints/granite_geospatial_uki_flood_detection_v1.ckpt
       backbone_pretrain_img_size: 512
       decoder_channels: 256
       backbone_bands: