| 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 | |