torchgeo/tests/conf/so2sat_s1.yaml

model:
  class_path: ClassificationTask
  init_args:
    loss: 'focal'
    model: 'resnet18'
    in_channels: 8
    num_classes: 17
data:
  class_path: So2SatDataModule
  init_args:
    batch_size: 1
    band_set: 's1'
  dict_kwargs:
    root: 'tests/data/so2sat'
    version: '2'