experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[TF] Fix test_group_conv flakiness.

Browse Source 
Increase the error tolerance from rtol=1e-05 and atol=1e-06 to rtol=3e-5 and
atol=3e-5. This fixes the flakiness of the test. Enable the test.
PiperOrigin-RevId: 323443013
Change-Id: Ia36784ad44f7b6a2fa48a83435a66942646b3c48
This commit is contained in:
Bixia Zheng 2020-07-27 14:36:31 -07:00 committed by TensorFlower Gardener
parent 3f4e18fd93
commit ac4bda59d6
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/keras/layers/convolutional_test.py
View File

@ -433,7 +433,7 @@ class GroupedConvTest(keras_parameterized.TestCase):
('Conv2D', keras.layers.Conv2D, (32, 12, 12, 32)), ('Conv2D', keras.layers.Conv2D, (32, 12, 12, 32)),
('Conv3D', keras.layers.Conv3D, (32, 12, 12, 12, 32)), ('Conv3D', keras.layers.Conv3D, (32, 12, 12, 12, 32)),
) )
def disable_test_group_conv(self, layer_cls, input_shape): def test_group_conv(self, layer_cls, input_shape):
if test.is_gpu_available(cuda_only=True): if test.is_gpu_available(cuda_only=True):
with testing_utils.use_gpu(): with testing_utils.use_gpu():
inputs = random_ops.random_uniform(shape=input_shape) inputs = random_ops.random_uniform(shape=input_shape)
@ -448,8 +448,8 @@ class GroupedConvTest(keras_parameterized.TestCase):
for inputs, weights in zip(input_slices, weight_slices) for inputs, weights in zip(input_slices, weight_slices)
], ],
axis=-1) axis=-1)
self.assertAllClose(
self.assertAllClose(layer(inputs), expected_outputs, rtol=1e-5) layer(inputs), expected_outputs, rtol=3e-5, atol=3e-5)
def test_group_conv_depthwise(self): def test_group_conv_depthwise(self):
if test.is_gpu_available(cuda_only=True): if test.is_gpu_available(cuda_only=True):