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