From 713dd6f9eb4a9ec4396de92f3a0f55cc7fd5b94f Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Fri, 13 May 2022 04:07:41 +0000 Subject: [PATCH] update pp_hgnet.py --- ppcls/arch/backbone/legendary_models/pp_hgnet.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/ppcls/arch/backbone/legendary_models/pp_hgnet.py b/ppcls/arch/backbone/legendary_models/pp_hgnet.py index 253783a9d..3e0412dfb 100644 --- a/ppcls/arch/backbone/legendary_models/pp_hgnet.py +++ b/ppcls/arch/backbone/legendary_models/pp_hgnet.py @@ -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)