/
githubmirror
/
transformers
Обзор
Документация
Войти
/
githubmirror
/
transformers
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
benchmark/optimum_benchmark_wrapper.py
20 строк
656 B
Yuanyuan Chen
Enable ruff on benchmark and scripts (#40634)
10 сен 2025, 12:38
Не верифицирован
10 сен 2025, 12:38
a5ecd94
Код
Авторство
О чём код?
import argparse import subprocess def main(config_dir, config_name, args): subprocess.run( ["optimum-benchmark", "--config-dir", f"{config_dir}", "--config-name", f"{config_name}"] + ["hydra/job_logging=disabled", "hydra/hydra_logging=disabled"] + args ) if __name__ == "__main__": parser = argparse.ArgumentParser() parser.add_argument("--config-dir", type=str, required=True, help="The path to the config directory.") parser.add_argument("--config-name", type=str, required=True, help="The config name.") args, unknown = parser.parse_known_args() main(args.config_dir, args.config_name, unknown)