Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
f6df698c
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
接近 2 年 前同步成功
通知
116
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看板
提交
f6df698c
编写于
1月 06, 2023
作者:
G
gaotingquan
提交者:
Tingquan Gao
1月 06, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
micro_block -> layer_type
上级
81de331e
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
14 addition
and
14 deletion
+14
-14
ppcls/arch/backbone/legendary_models/resnet.py
ppcls/arch/backbone/legendary_models/resnet.py
+13
-13
ppcls/configs/ImageNet/ResNet/ResNet18_dbb.yaml
ppcls/configs/ImageNet/ResNet/ResNet18_dbb.yaml
+1
-1
未找到文件。
ppcls/arch/backbone/legendary_models/resnet.py
浏览文件 @
f6df698c
...
...
@@ -164,18 +164,18 @@ class BottleneckBlock(TheseusLayer):
stride
,
shortcut
=
True
,
if_first
=
False
,
micro_block
=
ConvBNLayer
,
layer
=
ConvBNLayer
,
lr_mult
=
1.0
,
data_format
=
"NCHW"
):
super
().
__init__
()
self
.
conv0
=
micro_block
(
self
.
conv0
=
layer
(
num_channels
=
num_channels
,
num_filters
=
num_filters
,
filter_size
=
1
,
act
=
"relu"
,
lr_mult
=
lr_mult
,
data_format
=
data_format
)
self
.
conv1
=
micro_block
(
self
.
conv1
=
layer
(
num_channels
=
num_filters
,
num_filters
=
num_filters
,
filter_size
=
3
,
...
...
@@ -183,7 +183,7 @@ class BottleneckBlock(TheseusLayer):
act
=
"relu"
,
lr_mult
=
lr_mult
,
data_format
=
data_format
)
self
.
conv2
=
micro_block
(
self
.
conv2
=
layer
(
num_channels
=
num_filters
,
num_filters
=
num_filters
*
4
,
filter_size
=
1
,
...
...
@@ -226,13 +226,13 @@ class BasicBlock(TheseusLayer):
stride
,
shortcut
=
True
,
if_first
=
False
,
micro_block
=
ConvBNLayer
,
layer
=
ConvBNLayer
,
lr_mult
=
1.0
,
data_format
=
"NCHW"
):
super
().
__init__
()
self
.
stride
=
stride
self
.
conv0
=
micro_block
(
self
.
conv0
=
layer
(
num_channels
=
num_channels
,
num_filters
=
num_filters
,
filter_size
=
3
,
...
...
@@ -240,7 +240,7 @@ class BasicBlock(TheseusLayer):
act
=
"relu"
,
lr_mult
=
lr_mult
,
data_format
=
data_format
)
self
.
conv1
=
micro_block
(
self
.
conv1
=
layer
(
num_channels
=
num_filters
,
num_filters
=
num_filters
,
filter_size
=
3
,
...
...
@@ -296,7 +296,7 @@ class ResNet(TheseusLayer):
input_image_channel
=
3
,
return_patterns
=
None
,
return_stages
=
None
,
micro_block
=
"ConvBNLayer"
,
layer_type
=
"ConvBNLayer"
,
use_first_short_conv
=
True
,
**
kargs
):
super
().
__init__
()
...
...
@@ -312,10 +312,10 @@ class ResNet(TheseusLayer):
self
.
num_channels
=
self
.
cfg
[
"num_channels"
]
self
.
channels_mult
=
1
if
self
.
num_channels
[
-
1
]
==
256
else
4
if
micro_block
==
"ConvBNLayer"
:
micro_block
=
ConvBNLayer
elif
micro_block
==
"DiverseBranchBlock"
:
micro_block
=
DiverseBranchBlock
if
layer_type
==
"ConvBNLayer"
:
layer
=
ConvBNLayer
elif
layer_type
==
"DiverseBranchBlock"
:
layer
=
DiverseBranchBlock
else
:
raise
Exception
()
...
...
@@ -377,7 +377,7 @@ class ResNet(TheseusLayer):
if
i
==
0
and
block_idx
!=
0
else
1
,
shortcut
=
shortcut
,
if_first
=
block_idx
==
i
==
0
if
version
==
"vd"
else
True
,
micro_block
=
micro_block
,
layer
=
layer
,
lr_mult
=
self
.
lr_mult_list
[
block_idx
+
1
],
data_format
=
data_format
))
shortcut
=
True
...
...
ppcls/configs/ImageNet/ResNet/ResNet18_dbb.yaml
浏览文件 @
f6df698c
...
...
@@ -19,7 +19,7 @@ Global:
Arch
:
name
:
ResNet18
class_num
:
1000
micro_block
:
DiverseBranchBlock
layer_type
:
DiverseBranchBlock
use_first_short_conv
:
False
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录