diff --git a/python/paddle/fluid/tests/unittests/seresnext_net.py b/python/paddle/fluid/tests/unittests/seresnext_net.py index 252ffe2c22ad2e9d0b48e89d9dc0a7d695e1c0a4..17e0cd0d5b18652f828af9936b07cb4122f87b97 100644 --- a/python/paddle/fluid/tests/unittests/seresnext_net.py +++ b/python/paddle/fluid/tests/unittests/seresnext_net.py @@ -22,6 +22,7 @@ from simple_nets import init_data import math import os os.environ['CPU_NUM'] = str(4) +os.environ['FLAGS_cudnn_deterministic'] = str(1) # FIXME(zcd): If the neural net has dropout_op, the output of ParallelExecutor # and Executor is different. Because, for ParallelExecutor, the dropout_op of @@ -35,14 +36,10 @@ remove_dropout = False # and Executor is different. remove_bn = False -# FIXME(huihuangzheng): Temporarily disable cudnn of conv2d in unit test because -# it will cause random test failure. We have to re-enable it after someone fixs -# cudnn_conv remove_cudnn_conv = False remove_dropout = True remove_bn = True -remove_cudnn_conv = True def squeeze_excitation(input, num_channels, reduction_ratio):