Download src/bench/eval_pt.py from Aryan006/cone-distance: direct link, hf CLI and curl.
- Browser
- Download file 492 Bytes
-
https://huggingface.co/Aryan006/cone-distance/resolve/main/src/bench/eval_pt.py
- Command line
-
hf download hf://Aryan006/cone-distance/src/bench/eval_pt.py
-
curl -L -o eval_pt.py https://huggingface.co/Aryan006/cone-distance/resolve/main/src/bench/eval_pt.py
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") | |