工具说明
测试性能、输出结果,可检查与Caffe / Tensorflow的预期结果是否匹配。
参数
- 第一个参数指定 待测试模型的二进制文件。
- 第二个参数指定 性能测试的循环次数,10就表示循环推理10次。
- 第三个参数指定 是否输出推理中间结果,0为不输出,1为输出;默认输出当前目录的output目录下。
- 第四个参数指定 执行推理的计算设备,有效值为 0(浮点 CPU)、1(Metal)、3(浮点OpenCL)、6(OpenGL),7(Vulkan)
- 第五个参数指定 输入tensor的大小,一般不需要指定。
- 其他
默认输入与输出
功能
检查两个tensor文本文件是否一致。
- 0.1 表示绝对阈值,不输入则为 0.0001
- 比对值超过绝对阈值时,会直接输出到控制台
功能
参数
- 1 表示绝对阈值,不输入则为 0.0001
功能
Op 总耗时统计工具和模型运算量估计。
- 第一个参数 指定模型文件名
- 第二个参数 指定运行次数,默认 100
- 第三个参数 指定 执行推理的计算设备,有效值为 0(浮点 CPU)、1(Metal)、3(浮点OpenCL)、6(OpenGL),7(Vulkan)。(当执行推理的计算设备不为 CPU 时,Op 平均耗时和耗时占比可能不准)
- 第四个参数 指定输入大小,一般可不设
输出
- 第一列为 Op类型
- 第二列为 平均耗时
- 第三列为 耗时占比
- 示例:
功能
参数
- 该工具默认读取当前目录下的 input_0.txt 作为输入
- 第一个参数:模型文件
- 第二个参数:执行推理的计算设备