From 7f0075e8d03ffb2678dba26306da7723e5010477 Mon Sep 17 00:00:00 2001 From: zhiboniu Date: Fri, 27 May 2022 07:54:26 +0000 Subject: [PATCH] reverse to BatchNorm --- ppcls/arch/backbone/legendary_models/resnet.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ppcls/arch/backbone/legendary_models/resnet.py b/ppcls/arch/backbone/legendary_models/resnet.py index 88b197c8..4a3d40f3 100644 --- a/ppcls/arch/backbone/legendary_models/resnet.py +++ b/ppcls/arch/backbone/legendary_models/resnet.py @@ -137,11 +137,11 @@ class ConvBNLayer(TheseusLayer): weight_attr = ParamAttr(learning_rate=lr_mult, trainable=True) bias_attr = ParamAttr(learning_rate=lr_mult, trainable=True) - self.bn = BatchNorm2D( + self.bn = BatchNorm( num_filters, - weight_attr=ParamAttr(learning_rate=lr_mult), + param_attr=ParamAttr(learning_rate=lr_mult), bias_attr=ParamAttr(learning_rate=lr_mult), - data_format=data_format) + data_layout=data_format) self.relu = nn.ReLU() def forward(self, x): -- GitLab