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

6ee18c2b · Feng Ni · GitHub · dc6ba2ac · 6ee18c2b
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppdet/modeling/reid/resnet.py ppdet/modeling/reid/resnet.py +1 -1

未找到文件。
--- 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):