提交 713dd6f9 编写于 作者: C cuicheng01

update pp_hgnet.py

上级 8a760fb8
......@@ -90,7 +90,7 @@ class ESEModule(TheseusLayer):
return paddle.multiply(x=identity, y=x)
class _HG_Block(TheseusLayer):
class HG_Block(TheseusLayer):
def __init__(
self,
in_channels,
......@@ -140,7 +140,7 @@ class _HG_Block(TheseusLayer):
return x
class _HG_Stage(TheseusLayer):
class HG_Stage(TheseusLayer):
def __init__(self,
in_channels,
mid_channels,
......@@ -161,7 +161,7 @@ class _HG_Stage(TheseusLayer):
blocks_list = []
blocks_list.append(
_HG_Block(
HG_Block(
in_channels,
mid_channels,
out_channels,
......@@ -169,7 +169,7 @@ class _HG_Stage(TheseusLayer):
identity=False))
for _ in range(block_num - 1):
blocks_list.append(
_HG_Block(
HG_Block(
out_channels,
mid_channels,
out_channels,
......@@ -228,7 +228,7 @@ class PPHGNet(TheseusLayer):
in_channels, mid_channels, out_channels, block_num, downsample = stage_config[
k]
self.stages.append(
_HG_Stage(in_channels, mid_channels, out_channels, block_num,
HG_Stage(in_channels, mid_channels, out_channels, block_num,
layer_num, downsample))
self.avg_pool = AdaptiveAvgPool2D(1)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册