wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-27 23:23:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix (#1174)

Browse Source
This commit is contained in:
Eric Liang
2017-11-01 13:45:39 -07:00
committed by Philipp Moritz
parent dc66a2d7d5
commit 202e7bf19a
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/parallel.py
+6 -2
View File
@@ -65,8 +65,12 @@ class LocalSyncParallelOptimizer(object):
# Then setup the per-device loss graphs that use the shared weights
self._batch_index = tf.placeholder(tf.int32)
data_splits = zip(
*[tf.split(ph, len(devices)) for ph in input_placeholders])
# Split on the CPU in case the data doesn't fit in GPU memory.
with tf.device("/cpu:0"):
data_splits = zip(
*[tf.split(ph, len(devices)) for ph in input_placeholders])
self._towers = []
for device, device_placeholders in zip(self.devices, data_splits):
self._towers.append(self._setup_device(device,