Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
慢慢CG
Mace
提交
3270a015
Mace
项目概览
慢慢CG
/
Mace
与 Fork 源项目一致
Fork自
Xiaomi / Mace
通知
1
Star
0
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看板
提交
3270a015
编写于
1月 19, 2018
作者:
Y
yejianwu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
move validate_model.sh's params to model.config
上级
ff10ba59
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
10 addition
and
11 deletion
+10
-11
model.config
model.config
+8
-2
validate_model.sh
validate_model.sh
+2
-9
未找到文件。
model.config
浏览文件 @
3270a015
TF_INPUT_NODE
=
input
TF_OUTPUT_NODE
=
softmax
/
Reshape_1
TF_OUTPUT_BR_NODE
=
GCN
/
br_result_2
/
fcn_br
\ No newline at end of file
TF_OUTPUT_NODE
=
output
TF_OUTPUT_BR_NODE
=
before_softmax_node
TF_MODEL_FILE_PATH
=
path
/
to
/
tf_model_opt
.
pb
MODEL_TAG
=
mobile_net
INPUT_SHAPE
=
1
,
64
,
64
,
3
OUTPUT_SHAPE
=
1
,
64
,
64
,
2
RUNTIME
=
gpu
TUNING_OR_NOT
=
1
validate_model.sh
浏览文件 @
3270a015
...
...
@@ -2,23 +2,16 @@
# Must run at root dir of mace project.
set
+x
Usage
()
{
echo
'Usage: bash tools/validate_model.sh tools/model.config
tf_model_path model_tag input_shape output_shape runtime[gpu/dsp] [tuning]
'
echo
'Usage: bash tools/validate_model.sh tools/model.config'
}
if
[
$#
-lt
6
]
;
then
if
[
$#
-lt
1
]
;
then
Usage
exit
-1
fi
source
$1
TF_MODEL_FILE_PATH
=
$2
MODEL_TAG
=
$3
INPUT_SHAPE
=
$4
OUTPUT_SHAPE
=
$5
RUNTIME
=
$6
TUNING_OR_NOT
=
${
7
:-
0
}
if
[
x
"
$RUNTIME
"
=
x
"dsp"
]
;
then
DATA_TYPE
=
"DT_UINT8"
DEVICE_TYPE
=
"HEXAGON"
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录