wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-09 21:51:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

temp fix for build (#5006)

Browse Source
This commit is contained in:
Eric Liang
2019-06-20 18:07:44 -07:00
committed by Philipp Moritz
parent e59e8074dd
commit 1d17125333
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/tests/test_optimizers.py
+5 -5
View File
@@ -125,14 +125,14 @@ class AsyncSamplesOptimizerTest(unittest.TestCase):
def testMultiGPU(self):
local, remotes = self._make_evs()
workers = WorkerSet._from_existing(local, remotes)
optimizer = AsyncSamplesOptimizer(workers, num_gpus=2, _fake_gpus=True)
optimizer = AsyncSamplesOptimizer(workers, num_gpus=1, _fake_gpus=True)
self._wait_for(optimizer, 1000, 1000)
def testMultiGPUParallelLoad(self):
local, remotes = self._make_evs()
workers = WorkerSet._from_existing(local, remotes)
optimizer = AsyncSamplesOptimizer(
workers, num_gpus=2, num_data_loader_buffers=2, _fake_gpus=True)
workers, num_gpus=1, num_data_loader_buffers=1, _fake_gpus=True)
self._wait_for(optimizer, 1000, 1000)
def testMultiplePasses(self):
@@ -211,21 +211,21 @@ class AsyncSamplesOptimizerTest(unittest.TestCase):
num_data_loader_buffers=2, minibatch_buffer_size=4))
optimizer = AsyncSamplesOptimizer(
workers,
num_gpus=2,
num_gpus=1,
train_batch_size=100,
sample_batch_size=50,
_fake_gpus=True)
self._wait_for(optimizer, 1000, 1000)
optimizer = AsyncSamplesOptimizer(
workers,
num_gpus=2,
num_gpus=1,
train_batch_size=100,
sample_batch_size=25,
_fake_gpus=True)
self._wait_for(optimizer, 1000, 1000)
optimizer = AsyncSamplesOptimizer(
workers,
num_gpus=2,
num_gpus=1,
train_batch_size=100,
sample_batch_size=74,
_fake_gpus=True)