diff --git a/configs/segformer_config_RGB.yaml b/configs/segformer_config_RGB.yaml index 04c5e54a..a5a07c1e 100644 --- a/configs/segformer_config_RGB.yaml +++ b/configs/segformer_config_RGB.yaml @@ -57,6 +57,7 @@ model: class_path: segmentation_models_pytorch.losses.DiceLoss init_args: mode: "multiclass" + freeze_encoder: true class_labels: ["background", "fore", "hydro", "roads", "buildings"] class_colors: ["#000000", "#008000", "#0000FF", "#FFFF00", "#FF0000"] weights_from_checkpoint_path: null