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

Disable some Keras + TPUStrategy tests because they are flaky.

Browse Source 
PiperOrigin-RevId: 260180753
This commit is contained in:
Priya Gupta 2019-07-26 11:27:34 -07:00 committed by TensorFlower Gardener
parent afe9cf3cd2
commit d0181e13c8
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tensorflow/python/keras/distribute/distribute_strategy_test.py
View File

@ -1511,8 +1511,14 @@ class TestDistributionStrategyWithKerasModels(test.TestCase,
@combinations.generate(
combinations.times(all_strategy_combinations_minus_default(),
combinations.combine(run_distributed=[True, False])))
def test_distribution_strategy_with_symbolic_add_loss(self, distribution,
run_distributed):
def test_distribution_strategy_with_symbolic_add_loss(
self, mode, distribution, run_distributed):
# TODO(b/123533246): Enable the test for TPU once bug is fixed
if (isinstance(distribution,
(tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)) and
mode == 'graph' and not run_distributed):
self.skipTest('TPU Strategy in graph mode fails with this test.')
def _make_model_with_add_loss():
inputs = keras.Input((10,))
@ -1840,6 +1846,11 @@ class TestDistributionStrategyWithMultipleAddLossAndMetricCalls(
l1=[0.01],
l2=[0.1])))
def test_fit_and_evaluate(self, distribution, model_fn, l1, l2):
# TODO(b/138445028): Enable the test for TPU once bug is fixed.
if (isinstance(distribution,
(tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1))):
self.skipTest('Flaky with TPUStrategy')
# Make fake MNIST-like image data.
dataset = dataset_ops.DatasetV2.from_tensor_slices(
(np.random.uniform(size=(64, 28, 28, 1)).astype(np.float32),