fixed bugs in batchnorm testcase (#2995)

PaddlePaddle · Nov 25, 2024 · 1c95e07 · 1c95e07
1 parent 21cb73b
commit 1c95e07
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_8_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm1D_8_class.py
@@ -10,7 +10,7 @@ class LayerCase(paddle.nn.Layer):
 
     def __init__(self):
         super(LayerCase, self).__init__()
-        self.func = paddle.nn.BatchNorm1D(num_features=1, momentum=0.1, epsilon=1e-05, weight_attr=False, data_format='NCL', )
+        self.func = paddle.nn.BatchNorm1D(num_features=1, momentum=0.1, epsilon=1e-05, weight_attr=False, bias_attr=False, data_format='NCL', )
 
     def forward(self, data, ):
         """

diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_7_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm2D_7_class.py
@@ -10,7 +10,7 @@ class LayerCase(paddle.nn.Layer):
 
     def __init__(self):
         super(LayerCase, self).__init__()
-        self.func = paddle.nn.BatchNorm2D(num_features=1, momentum=0.1, epsilon=1e-05, data_format='NCHW', weight_attr=False, )
+        self.func = paddle.nn.BatchNorm2D(num_features=1, momentum=0.1, epsilon=1e-05, data_format='NCHW', weight_attr=False, bias_attr=False)
 
     def forward(self, data, ):
         """

diff --git a/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_4_class.py b/framework/e2e/PaddleLT_new/layerApicase/nn_sublayer/BatchNorm3D_4_class.py
@@ -10,7 +10,7 @@ class LayerCase(paddle.nn.Layer):
 
     def __init__(self):
         super(LayerCase, self).__init__()
-        self.func = paddle.nn.BatchNorm3D(num_features=1, momentum=0.1, epsilon=1e-05, weight_attr=False, )
+        self.func = paddle.nn.BatchNorm3D(num_features=1, momentum=0.1, epsilon=1e-05, weight_attr=False, bias_attr=False)
 
     def forward(self, data, ):
         """