diff --git a/ppdet/modeling/reid/resnet.py b/ppdet/modeling/reid/resnet.py index c2261e0d0776b2823633d52c384bf416314d1e0b..2e2a85558d69cecb307df1f1098ec0bdd70a93e2 100644 --- a/ppdet/modeling/reid/resnet.py +++ b/ppdet/modeling/reid/resnet.py @@ -55,7 +55,7 @@ class ConvBNLayer(nn.Layer): bias_attr=False, data_format=data_format) - self._batch_norm = nn.BatchNorm2D(num_filters, data_layout=data_format) + self._batch_norm = nn.BatchNorm2D(num_filters) self.act = act def forward(self, inputs):