Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
闹闹不闹!
Mace
提交
75531cad
Mace
项目概览
闹闹不闹!
/
Mace
与 Fork 源项目一致
Fork自
Xiaomi / Mace
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
Mace
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
75531cad
编写于
12月 17, 2018
作者:
B
Bin Li
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix quantize stat
上级
09a7b52d
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
6 addition
and
2 deletion
+6
-2
tools/device.py
tools/device.py
+6
-2
未找到文件。
tools/device.py
浏览文件 @
75531cad
...
...
@@ -175,7 +175,8 @@ class DeviceWrapper:
output_file_name
=
'model_out'
,
runtime_failure_ratio
=
0.0
,
address_sanitizer
=
False
,
link_dynamic
=
False
link_dynamic
=
False
,
quantize_stat
=
False
,
):
six
.
print_
(
"* Run '%s' with round=%s, restart_round=%s, tuning=%s, "
"out_of_range_check=%s, omp_num_threads=%s, "
...
...
@@ -196,6 +197,7 @@ class DeviceWrapper:
"LD_LIBRARY_PATH=%s"
%
libmace_dynamic_lib_path
,
"MACE_CPP_MIN_VLOG_LEVEL=%s"
%
vlog_level
,
"MACE_RUNTIME_FAILURE_RATIO=%f"
%
runtime_failure_ratio
,
"MACE_LOG_TENSOR_RANGE=%d"
%
(
1
if
quantize_stat
else
0
),
"%s/%s"
%
(
target_dir
,
target_name
),
"--model_name=%s"
%
model_tag
,
"--input_node=%s"
%
","
.
join
(
input_nodes
),
...
...
@@ -277,6 +279,7 @@ class DeviceWrapper:
"MACE_INTERNAL_STORAGE_PATH=%s"
%
internal_storage_dir
,
"MACE_LIMIT_OPENCL_KERNEL_TIME=%s"
%
limit_opencl_kernel_time
,
"MACE_RUNTIME_FAILURE_RATIO=%f"
%
runtime_failure_ratio
,
"MACE_LOG_TENSOR_RANGE=%d"
%
(
1
if
quantize_stat
else
0
),
]
if
self
.
system
==
SystemType
.
android
and
address_sanitizer
:
cmd
.
extend
([
...
...
@@ -542,7 +545,8 @@ class DeviceWrapper:
opencl_binary_file
=
model_opencl_output_bin_path
,
opencl_parameter_file
=
model_opencl_parameter_path
,
libmace_dynamic_library_path
=
LIBMACE_DYNAMIC_PATH
,
link_dynamic
=
link_dynamic
link_dynamic
=
link_dynamic
,
quantize_stat
=
flags
.
quantize_stat
,
)
if
flags
.
validate
:
model_file_path
,
weight_file_path
=
get_model_files
(
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录