Fix a Cudnn Deterministic Random Failure, test=develop (#25389)

As the title
fix_copy_if_different
Huihuang Zheng 5 years ago committed by GitHub
parent 6de75082cb
commit ff7af219f1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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):

Loading…
Cancel
Save