super-gradients/src/super_gradients/recipes/cityscapes_stdc_base.yaml

# STDC Base training params

defaults:
  - training_hyperparams: cityscapes_default_train_params
  - dataset_params: cityscapes_dataset_params
  - checkpoint_params: default_checkpoint_params
  - _self_

train_dataloader: cityscapes_train
val_dataloader: cityscapes_val

data_loader_num_workers: 10

arch_params:
  num_classes: 19
  use_aux_heads: True

checkpoint_params:
  checkpoint_path:
  load_backbone: True
  load_weights_only: True
  strict_load: no_key_matching

architecture: stdc1_seg
experiment_name: ${architecture}_cityscapes

training_hyperparams:
  sync_bn: True



ckpt_root_dir:

multi_gpu: DDP


# THE FOLLOWING PARAMS ARE DIRECTLY USED BY HYDRA
hydra:
  run:
    # Set the output directory (i.e. where .hydra folder that logs all the input params will be generated)
    dir: ${hydra_output_dir:${ckpt_root_dir}, ${experiment_name}}