Skip to content

Commit cea8373

Browse files
committed
add unittest
1 parent 2d36a20 commit cea8373

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

python/paddle/fluid/tests/unittests/test_batch_norm_op.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -515,6 +515,13 @@ def init_test_case(self):
515515
os.environ['FLAGS_cudnn_batchnorm_spatial_persistent'] = "1"
516516

517517

518+
class TestBatchNormOpTrainingCase3(TestBatchNormOpTraining):
519+
def init_test_case(self):
520+
self.use_global_stats = False
521+
self.no_grad_set = set(['x@GRAD'])
522+
self.fetch_list = ['y', 'mean', 'variance', 'scale@GRAD', 'bias@GRAD']
523+
524+
518525
class TestBatchNormOpTrainingMomentumVariable(TestBatchNormOpTraining):
519526
def init_test_case(self):
520527
self.use_momentum_variable = True

0 commit comments

Comments
 (0)