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