Add optional load_parts parameter to SegFormer configuration

NRCan · Mar 7, 2025 · 233951b · 233951b
1 parent 7247a27
commit 233951b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/configs/segformer_config_RGB.yaml b/configs/segformer_config_RGB.yaml
@@ -77,7 +77,7 @@ model:
     class_labels: ["background", "fore", "hydro", "roads", "buildings"]
     class_colors: ["#000000", "#008000", "#0000FF", "#FFFF00", "#FF0000"]
     weights_from_checkpoint_path: null
-
+    load_parts: null
 data:
   class_path: datamodules.imagery_NonGeoDataModule.BlueSkyNonGeoDataModule
   init_args: