bench.py

#!/usr/bin/python3
# -*- coding: UTF-8 -*-

from core.engine import Engine
from core.global_config import create_config


def main():
    config_dict = create_config("tnn")
    tnn = Engine(config_dict)
    model_dict = tnn.prepare_models()
    device_dict = tnn.prepare_devices()
    tnn.set_config("model_dict", model_dict)
    tnn.set_config("device_dict", device_dict)

    tnn.prepare_models_for_devices()
    tnn.prepare_benchmark_assets_for_devices()
    bench_dict = tnn.run_bench()
    bench_str_list = tnn.generate_benchmark_summary(bench_dict)
    tnn.write_list_to_file(bench_str_list)


if __name__ == "__main__":
    main()