benchmark.py 文件源码

def run_benchmark(input_data):
    print("Running dry runs...")
    for n in trange(args.dry_runs):
        iteration(input_data)

    print("\n Running measured runs...")
    running_time = 0
    for n in trange(args.runs):
        start, end = iteration(input_data)
        running_time += end - start

    return running_time / float(args.runs)