# Shared training config for Baseline A, B, C, and E3 PhysioJEPA.
# Only `run_name` and `model` differ across runs — everything else stays identical
# so K2 (E3 AUROC > Baseline B + 0.02) is a single-variable comparison.

run_name: base
model: F                 # override per-run: A|B|C|F
epochs: 100
batch_size: 64
lr: 1.0e-4
weight_decay: 0.04
warmup_epochs: 10
ema_start: 0.996
ema_end: 0.9999
ema_warmup_frac: 0.30
grad_clip: 1.0
log_every: 100
ckpt_every_epochs: 5
seed: 42
wandb_project: physiojepa
wandb_mode: online
wandb_entity: null
output_dir: runs
index_path: cache/mimic_index.json
shard_roots: []          # filled per-environment (populated by prepare_data.py)
num_workers: 4
amp: true
log_uniform_frac: 0.6