cone-distance / src /bench /eval_pt.py
Aryan Sethi
Claude Opus 5 (1M context)
Phase 3: per-format benchmark with stage timing
0b910fa
Raw History Blame Contribute Delete
492 Bytes
"""Benchmark the PyTorch checkpoint.
The reference point. Slowest of the three on CPU and the only one that can use
a GPU, so it is also the format to run with `--device 0` when there is one.
python -m src.bench.eval_pt
python -m src.bench.eval_pt --device 0 --limit 200
"""
from src.bench.common_cli import build_parser, run
if __name__ == "__main__":
parser = build_parser(__doc__, default_weights="yolo11n.pt")
run(parser.parse_args(), label="PyTorch (.pt)", tag="pt")