Spaces:
Running
Running
File size: 873 Bytes
cb44915 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
"""
CLI entry point for running evaluation as a module.
Usage:
python -m src --model ./my_model/final
python -m src --model username/model-name
"""
import argparse
import sys
from .evaluate import evaluate_model
def main():
parser = argparse.ArgumentParser(
description="Evaluate a chess model",
prog="python -m src",
)
parser.add_argument(
"--model",
"-m",
type=str,
required=True,
help="Path to model directory or HuggingFace model ID",
)
parser.add_argument(
"--quiet",
"-q",
action="store_true",
help="Suppress progress output",
)
args = parser.parse_args()
result = evaluate_model(args.model, verbose=not args.quiet)
print()
print(result.summary())
return 0
if __name__ == "__main__":
sys.exit(main())
|