From 1c95e07e897f744932a96055f9cc39f893e281f9 Mon Sep 17 00:00:00 2001 From: cubehan3 Date: Mon, 25 Nov 2024 19:37:12 +0800 Subject: [PATCH] fixed bugs in batchnorm testcase (#2995) --- .../layerApicase/nn_sublayer/BatchNorm1D_8_class.py | 2 +- .../layerApicase/nn_sublayer/BatchNorm2D_7_class.py | 2 +- .../layerApicase/nn_sublayer/BatchNorm3D_4_class.py | 2 +- 3 files changed, 3 insertions(+), 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 index d810a20ff4..3f3f245f19 100644 --- 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 index 7df8f3b26a..f8bef6cffc 100644 --- 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 index a0b85c2fcd..c59cc798bd 100644 --- 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, ): """