more un-mpying

2018-11-09 10:08:39 -08:00
parent 84323c3d49
commit 0b8126f949
1 changed files with 3 additions and 2 deletions
--- a/baselines/ppo2/model.py
+++ b/baselines/ppo2/model.py
@@ -124,6 +124,7 @@ class Model(object):

        if MPI is None or MPI.COMM_WORLD.Get_rank() == 0:
            initialize()
+        else:
            global_variables = tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, scope="")
            sync_from_root(sess, global_variables) #pylint: disable=E1101