"...git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "92b24d97c9211a4156dbeabf0a7e2ce9bf348f5e"
Commit 4d4eb85c authored by josh11b's avatar josh11b Committed by Taylor Robie
Browse files

num_replicas -> num_replicas_in_sync (#5752)

`num_replicas` is being replaced by `num_replicas_in_sync` and something that returns the number of steps running concurrently.
parent 628b970a
...@@ -27,19 +27,19 @@ class GetDistributionStrategyTest(tf.test.TestCase): ...@@ -27,19 +27,19 @@ class GetDistributionStrategyTest(tf.test.TestCase):
"""Tests for get_distribution_strategy.""" """Tests for get_distribution_strategy."""
def test_one_device_strategy_cpu(self): def test_one_device_strategy_cpu(self):
ds = distribution_utils.get_distribution_strategy(0) ds = distribution_utils.get_distribution_strategy(0)
self.assertEquals(ds.num_replicas, 1) self.assertEquals(ds.num_replicas_in_sync, 1)
self.assertEquals(len(ds.worker_devices), 1) self.assertEquals(len(ds.worker_devices), 1)
self.assertIn('CPU', ds.worker_devices[0]) self.assertIn('CPU', ds.worker_devices[0])
def test_one_device_strategy_gpu(self): def test_one_device_strategy_gpu(self):
ds = distribution_utils.get_distribution_strategy(1) ds = distribution_utils.get_distribution_strategy(1)
self.assertEquals(ds.num_replicas, 1) self.assertEquals(ds.num_replicas_in_sync, 1)
self.assertEquals(len(ds.worker_devices), 1) self.assertEquals(len(ds.worker_devices), 1)
self.assertIn('GPU', ds.worker_devices[0]) self.assertIn('GPU', ds.worker_devices[0])
def test_mirrored_strategy(self): def test_mirrored_strategy(self):
ds = distribution_utils.get_distribution_strategy(5) ds = distribution_utils.get_distribution_strategy(5)
self.assertEquals(ds.num_replicas, 5) self.assertEquals(ds.num_replicas_in_sync, 5)
self.assertEquals(len(ds.worker_devices), 5) self.assertEquals(len(ds.worker_devices), 5)
for device in ds.worker_devices: for device in ds.worker_devices:
self.assertIn('GPU', device) self.assertIn('GPU', device)
......
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