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

Add MWMS combinations to custom_training_loop_metrics_test.

Browse Source 
PiperOrigin-RevId: 324726066
Change-Id: I05fcf563c34c216d9a59a1656e3ce4d409e2a8f3
This commit is contained in:
Anjali Sridhar 2020-08-03 18:26:43 -07:00 committed by TensorFlower Gardener
parent 9e6c87150b
commit e2865bb150
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/python/keras/distribute/custom_training_loop_metrics_test.py
View File

@ -34,7 +34,8 @@ class KerasMetricsTest(test.TestCase, parameterized.TestCase):
@combinations.generate(
combinations.combine(
distribution=strategy_combinations.all_strategies,
distribution=strategy_combinations.all_strategies +
strategy_combinations.multiworker_strategies,
mode=["eager"]
))
def test_multiple_keras_metrics_experimental_run(self, distribution):
@ -58,7 +59,8 @@ class KerasMetricsTest(test.TestCase, parameterized.TestCase):
@combinations.generate(
combinations.combine(
distribution=strategy_combinations.all_strategies,
distribution=strategy_combinations.all_strategies+
strategy_combinations.multiworker_strategies,
mode=["eager"]
))
def test_update_keras_metric_declared_in_strategy_scope(self, distribution):
@ -98,4 +100,4 @@ class KerasMetricsTest(test.TestCase, parameterized.TestCase):
if __name__ == "__main__":
test.main()
combinations.main()