Skip to content

GitLab

Commit 4b566d4e authored by Tayo Oguntebi's avatar Tayo Oguntebi Committed by Toby Boyd
Browse files

Updating stale DistributionStrategy test. (#6281)

parent e1ae37c4
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 6 deletions
official/utils/misc/distribution_utils_test.py
View file @ 4b566d4e
...@@ -44,12 +44,6 @@ class GetDistributionStrategyTest(tf.test.TestCase): ...@@ -44,12 +44,6 @@ class GetDistributionStrategyTest(tf.test.TestCase):
for device in ds.extended.worker_devices: for device in ds.extended.worker_devices:
self.assertIn('GPU', device) self.assertIn('GPU', device)
def test_override_strategy(self):
ds = distribution_utils.get_distribution_strategy(
distribution_strategy='collective', num_gpus=2)
self.assertTrue(
isinstance(ds, tf.contrib.distribute.CollectiveAllReduceStrategy))
class PerDeviceBatchSizeTest(tf.test.TestCase): class PerDeviceBatchSizeTest(tf.test.TestCase):
"""Tests for per_device_batch_size.""" """Tests for per_device_batch_size."""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment