pretrain_checkpoint: /path/to/checkpoint_best.pt
from_scratch: false
use_checkpoint_config: true
model:
  model_type: mamba
  num_classes: 3
  head_type: transformer
  head_depth: 2
  head_num_heads: 8
  head_mlp_ratio: 4.0
  head_proj_drop: 0.1
  head_drop_path: 0.1
  mlp_hidden: 512
  mlp_depth: 4
  mlp_dropout: 0.1
  freeze_backbone: false
data:
  train_list: data/lists/downstream_train.txt
  val_list: data/lists/downstream_val.txt
  test_list: data/lists/downstream_test.txt
  csv: data/labels.csv
  id_column: Subject
  label_column: Group_idx
  label_mode: multiclass
  path_id_mode: auto
  batch_size: 8
  num_workers: 8
  T_prime: 30
  tau_seconds: 6.0
  default_tr: null
training:
  epochs: 30
  lr: 1.0e-5
  lr_backbone: 6.0e-6
  lr_head: 6.0e-5
  weight_decay: 0.05
  warmup_epochs: 2
  grad_accumulation_steps: 2
  grad_clip: 1.0
  seed: 42
  use_amp: true
logging:
  log_interval: 20
  checkpoint_dir: ./checkpoints/downstream/example
  log_dir: ./logs/downstream/example