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")