NAME: SEnSeIv2-DeepLabv3-S2-unambiguous

#-------------
# Model options
#-------------
PATCH_SIZE: 512
SEnSeIv2: 'hf_models/sensei-configs/senseiv2-medium.yaml'
MODEL_TYPE: 'DeepLabv3+'
DEEPLAB_CONFIG: 'resnet50'
RECOVERY_MODULE: false
CLASSES: 4
MULTIMODAL: false
NUM_CHANNELS: null # Set to null for sensor independent models

#----------------
#training options (not needed for inference)
#----------------
EPOCHS: 70
BATCH_SIZE: 8
PHASES: [0, 1, 2, 45, 60, 65]
ACCUMULATE_STEPS: [1, 1, 1, 1, 4, 8]
LR: [0.000005, 0.00002, 0.0001, 0.00001, 0.00001, 0.000002]
EPSILON: 0.000001
WEIGHT_DECAY: 0.0001
L1_REG: 0
RECOVERY_WARMUP_STEPS: 5000
RECOVERY_LOSS_FACTOR: 1
LOSS: 'ambiguous_crossentropy_loss'


#Data options
TRAIN_DIRS:
  - 'path/to/be/set'
VALID_DIRS:
  - 'path/to/be/set'
MIN_BANDS: 3
MAX_BANDS: 13