def main(): """Main runner for shell""" from ai_benchmark import AIBenchmark parsed_args = parser.parse_known_args()[0] benchmark = AIBenchmark( use_cpu=parsed_args.use_cpu, verbose_level=parsed_args.verbose, seed=parsed_args.seed, ) test_info, results = benchmark.run( precision=parsed_args.precision, test_ids=parsed_args.test_ids, training=parsed_args.run_training, inference=parsed_args.run_inference, micro=parsed_args.run_micro, cpu_cores=parsed_args.cpu_cores, inter_threads=parsed_args.inter_threads, intra_threads=parsed_args.intra_threads, ) if parsed_args.json: output = vars(results) output['test_results'] = { k: vars(v) for k, v in output['test_results'].items() } output['test_info'] = vars(test_info) output['test_info'].pop('results', None) print(json.dumps(output, indent=4))
from ai_benchmark import AIBenchmark benchmark = AIBenchmark(use_CPU=True, verbose_level=3) results = benchmark.run(precision="high")
import tensorflow as tf print(tf.__version__) from ai_benchmark import AIBenchmark benchmark = AIBenchmark() results = benchmark.run()
from ai_benchmark import AIBenchmark teste = AIBenchmark(use_CPU=True) teste.run(precision="high")