Skip to content

SWEEP_README.md

Latest commit

 

History

History
19 lines (15 loc) · 740 Bytes

SWEEP_README.md

File metadata and controls

19 lines (15 loc) · 740 Bytes

Backbone sweep and ensemble

Run short pilots (backbone sweep)

python scripts/sweep_backbones.py --config configs/rtx6000_ada_train.yaml --epochs 10 \
  --backbones vit_base_patch16_224 eva02_base_patch14_224 convnext_large \
  --out_root runs/sweep --seed 42

After it finishes, see runs/sweep/summary.json for metrics per backbone.

Train the best backbone longer

Update configs/default.yaml (or a new config) with the chosen model.backbone_name and run the usual trainer.

Ensemble two+ checkpoints (logit average)

python scripts/ensemble_evaluate.py --config configs/rtx6000_ada_eval_heavy.yaml \
  --checkpoints runs/sweep/vit_base_patch16_224/best.pt runs/sweep/eva02_base_patch14_224/best.pt