diff --git a/ppcls/arch/backbone/legendary_models/mobilenet_v1.py b/ppcls/arch/backbone/legendary_models/mobilenet_v1.py index c8bc594f8ac527e245ef3f1711b374dfcf149e44..cf57f9882745e438a6d31ff7bb70e5ea7c282725 100644 --- a/ppcls/arch/backbone/legendary_models/mobilenet_v1.py +++ b/ppcls/arch/backbone/legendary_models/mobilenet_v1.py @@ -53,12 +53,10 @@ class ConvBNLayer(TheseusLayer): stride=stride, padding=padding, groups=num_groups, - weight_attr=ParamAttr( - initializer=KaimingNormal()), + weight_attr=ParamAttr(initializer=KaimingNormal()), bias_attr=False) - self.bn = BatchNorm( - num_filters) + self.bn = BatchNorm(num_filters) self.relu = ReLU()