From ff7af219f19a57dd03001767ca34ff7d261ad709 Mon Sep 17 00:00:00 2001 From: Huihuang Zheng Date: Tue, 7 Jul 2020 10:33:00 +0800 Subject: [PATCH] Fix a Cudnn Deterministic Random Failure, test=develop (#25389) As the title --- python/paddle/fluid/tests/unittests/seresnext_net.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/seresnext_net.py b/python/paddle/fluid/tests/unittests/seresnext_net.py index 252ffe2c22a..17e0cd0d5b1 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): -- GitLab