| seed_everything: 42 | |
| trainer: | |
| accelerator: auto | |
| strategy: auto | |
| devices: 1 | |
| max_epochs: 100 | |
| default_root_dir: ./checkpoints_s2_simulated | |
| callbacks: | |
| - class_path: lightning.pytorch.callbacks.ModelCheckpoint | |
| init_args: | |
| monitor: val/loss | |
| mode: min | |
| save_top_k: 1 | |
| filename: "best_model" | |
| save_last: true | |
| - class_path: lightning.pytorch.callbacks.LearningRateMonitor | |
| init_args: | |
| logging_interval: epoch | |
| model: | |
| class_path: terratorch.tasks.ClassificationTask | |
| init_args: | |
| model_factory: EncoderDecoderFactory | |
| loss: ce | |
| ignore_index: -1 | |
| lr: 1.0e-5 # Top-level LR | |
| optimizer: AdamW | |
| optimizer_hparams: | |
| weight_decay: 0.05 | |
| scheduler: ReduceLROnPlateau | |
| scheduler_hparams: | |
| mode: min | |
| patience: 5 | |
| model_args: | |
| backbone: terramind_v1_base | |
| backbone_pretrained: true | |
| backbone_modalities: | |
| - S2L2A | |
| backbone_merge_method: mean | |
| decoder: UperNetDecoder | |
| decoder_scale_modules: true | |
| decoder_channels: 256 | |
| num_classes: 2 | |
| head_dropout: 0.3 | |
| necks: | |
| - name: ReshapeTokensToImage | |
| remove_cls_token: false | |
| - name: SelectIndices | |
| indices: [2, 5, 8, 11] | |
| data: | |
| class_path: methane_simulated_datamodule.MethaneSimulatedDataModule | |
| init_args: | |
| data_root: /path/to/data_root # UPDATE THIS | |
| excel_file: ../../Methane_benchmark_patches_summary_v3.xlsx | |
| batch_size: 8 | |
| val_split: 0.2 | |
| seed: 42 | |
| test_fold: 4 | |
| num_folds: 5 | |
| sim_tag: toarefl # Change to 'boarefl' if needed | |