From a16de6da0fc34664b00e8f2367438d007020d666 Mon Sep 17 00:00:00 2001 From: weishengyu Date: Tue, 25 May 2021 16:55:38 +0800 Subject: [PATCH] remove name of conv and bn --- ppcls/arch/backbone/legendary_models/hrnet.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/ppcls/arch/backbone/legendary_models/hrnet.py b/ppcls/arch/backbone/legendary_models/hrnet.py index 2d6afad1..c094ba14 100644 --- a/ppcls/arch/backbone/legendary_models/hrnet.py +++ b/ppcls/arch/backbone/legendary_models/hrnet.py @@ -65,16 +65,11 @@ class ConvBNLayer(TheseusLayer): stride=stride, padding=(filter_size - 1) // 2, groups=groups, - weight_attr=ParamAttr(name=name + "_weights"), bias_attr=False) bn_name = name + '_bn' self._batch_norm = nn.BatchNorm( num_filters, - act=act, - param_attr=ParamAttr(name=bn_name + '_scale'), - bias_attr=ParamAttr(bn_name + '_offset'), - moving_mean_name=bn_name + '_mean', - moving_variance_name=bn_name + '_variance') + act=act) def forward(self, x, res_dict=None): y = self._conv(x) -- GitLab