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

[rllib] Fix PPO regression

Browse Source
This commit is contained in:
Richard Liaw
2018-06-28 16:00:53 -07:00
committed by Eric Liang
parent 89460b8d11
commit 92ab7e56ec
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/rllib/optimizers/multi_gpu_optimizer.py
+1 -2
View File
@@ -65,8 +65,7 @@ class LocalMultiGPUOptimizer(PolicyOptimizer):
# all of the device copies are created.
with self.local_evaluator.tf_sess.graph.as_default():
with self.local_evaluator.tf_sess.as_default():
main_scope = tf.get_variable_scope()
with tf.variable_scope(main_scope, reuse=tf.AUTO_REUSE):
with tf.variable_scope("default", reuse=tf.AUTO_REUSE):
self.par_opt = LocalSyncParallelOptimizer(
tf.train.AdamOptimizer(self.sgd_stepsize),
self.devices,