Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
毕竟曾有刹那
Mace
提交
83ccbd85
Mace
项目概览
毕竟曾有刹那
/
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
83ccbd85
编写于
5月 22, 2019
作者:
B
Bin Li
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix check tensor
上级
23d985f7
变更
1
显示空白变更内容
内联
并排
Showing
1 changed file
with
4 addition
and
14 deletion
+4
-14
mace/python/tools/converter_tool/transformer.py
mace/python/tools/converter_tool/transformer.py
+4
-14
未找到文件。
mace/python/tools/converter_tool/transformer.py
浏览文件 @
83ccbd85
...
...
@@ -137,7 +137,6 @@ class Transformer(base_converter.ConverterInterface):
changed
=
transformer
()
if
not
changed
:
break
self
.
delete_after_check_nodes
()
return
self
.
_model
,
self
.
_quantize_activation_info
def
initialize_name_map
(
self
):
...
...
@@ -1332,7 +1331,10 @@ class Transformer(base_converter.ConverterInterface):
visited
=
set
()
sorted_nodes
=
[]
for
output_node
in
self
.
_option
.
output_nodes
:
output_nodes
=
self
.
_option
.
check_nodes
if
not
self
.
_quantize_activation_info
:
output_nodes
.
update
(
self
.
_option
.
output_nodes
)
for
output_node
in
output_nodes
:
mace_check
(
output_node
in
self
.
_producer
,
"output_tensor %s not existed in model"
%
output_node
)
self
.
sort_dfs
(
self
.
_producer
[
output_node
],
visited
,
sorted_nodes
)
...
...
@@ -2010,18 +2012,6 @@ class Transformer(base_converter.ConverterInterface):
arg
.
i
=
mace_pb2
.
GPU_IMAGE
if
self
.
_option
.
cl_mem_type
==
"image"
\
else
mace_pb2
.
GPU_BUFFER
def
delete_after_check_nodes
(
self
):
if
self
.
_option
.
check_nodes
!=
self
.
_option
.
output_nodes
:
mace_check
(
len
(
self
.
_option
.
check_nodes
)
==
1
,
"Only support one check node now."
)
check_node
=
None
for
i
in
six
.
moves
.
range
(
len
(
self
.
_model
.
op
)):
if
self
.
_model
.
op
[
i
].
output
[
0
]
in
self
.
_option
.
check_nodes
:
check_node
=
self
.
_model
.
op
[
i
]
del
self
.
_model
.
op
[
i
+
1
:]
break
mace_check
(
check_node
is
not
None
,
"check node not found."
)
def
transform_caffe_reshape_and_flatten
(
self
):
net
=
self
.
_model
for
op
in
net
.
op
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录