fix reid resnet bn export, test=document_fix (PaddlePaddle#5162)

an1018 · Jan 28, 2022 · 6ee18c2 · 6ee18c2
1 parent dc6ba2a
commit 6ee18c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppdet/modeling/reid/resnet.py b/ppdet/modeling/reid/resnet.py
@@ -55,7 +55,7 @@ def __init__(self,
             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):