Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
43fce425
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
1 年多 前同步成功
通知
115
Star
4999
Fork
1114
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
19
列表
看板
标记
里程碑
合并请求
6
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleClas
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
19
Issue
19
列表
看板
标记
里程碑
合并请求
6
合并请求
6
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
43fce425
编写于
1月 07, 2022
作者:
G
gaotingquan
提交者:
Tingquan Gao
1月 14, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix: change the returned result from dict to list
上级
3b9f6292
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
5 addition
and
5 deletion
+5
-5
ppcls/arch/backbone/base/theseus_layer.py
ppcls/arch/backbone/base/theseus_layer.py
+5
-5
未找到文件。
ppcls/arch/backbone/base/theseus_layer.py
浏览文件 @
43fce425
...
...
@@ -112,7 +112,7 @@ class TheseusLayer(nn.Layer):
if
not
isinstance
(
layer_name_pattern
,
list
):
layer_name_pattern
=
[
layer_name_pattern
]
h
andle_res_dict
=
{}
h
it_layer_pattern_list
=
[]
for
pattern
in
layer_name_pattern
:
# parse pattern to find target layer and its parent
layer_list
=
parse_pattern_str
(
pattern
=
pattern
,
parent_layer
=
self
)
...
...
@@ -133,8 +133,8 @@ class TheseusLayer(nn.Layer):
else
:
setattr
(
sub_layer_parent
,
sub_layer_name
,
new_sub_layer
)
h
andle_res_dict
[
pattern
]
=
new_sub_layer
return
h
andle_res_dic
t
h
it_layer_pattern_list
.
append
(
pattern
)
return
h
it_layer_pattern_lis
t
def
stop_after
(
self
,
stop_layer_name
:
str
)
->
bool
:
"""stop forward and backward after 'stop_layer_name'.
...
...
@@ -192,7 +192,7 @@ class TheseusLayer(nn.Layer):
handle_func
=
Handler
(
self
.
res_dict
)
res_dic
t
=
self
.
upgrade_sublayer
(
hit_layer_pattern_lis
t
=
self
.
upgrade_sublayer
(
return_patterns
,
handle_func
=
handle_func
)
if
hasattr
(
self
,
"hook_remove_helper"
):
...
...
@@ -200,7 +200,7 @@ class TheseusLayer(nn.Layer):
self
.
hook_remove_helper
=
self
.
register_forward_post_hook
(
self
.
_return_dict_hook
)
return
res_dic
t
return
hit_layer_pattern_lis
t
def
set_identity
(
parent_layer
:
nn
.
Layer
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录