本章节将通过一个简单的可执行示例帮助你快速上手
用户使用launch.py作为入口脚本。当使用byte mlperf评估模型时,您只需要传入两个参数:--task 和 --hardware_type,如下所示:
--task
: 此参数是传入的任务名称。您需要指定任务。例如,如果您想评估任务: bert-tf-fp16.json
,您需要指定 --task bert-tf-fp16
。
注意:所有的任务配置都定义在byte_mlperf/workloads
下,传递参数时名称需要与文件名对齐。当前的格式是模型-框架-精度。
--hardware_type
: 此参数是传入的硬件类型名称,没有默认值,用户必须指定。例如:要评估Habana Goya,请指定 --hardware_type GOYA
。
注意:所有的硬件类型都定义在byte_mlperf/backends
下,并且传递参数时名称需要与文件夹名称对齐。
--compile_only
: 此参数将在任务编译完成后停止。
--show_task_list
: 此参数将打印所有任务列表。
--show_hardware_list
: 此参数将打印所有支持的硬件列表。