more un-mpying
This commit is contained in:
@@ -124,8 +124,9 @@ class Model(object):
|
|||||||
|
|
||||||
if MPI is None or MPI.COMM_WORLD.Get_rank() == 0:
|
if MPI is None or MPI.COMM_WORLD.Get_rank() == 0:
|
||||||
initialize()
|
initialize()
|
||||||
global_variables = tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, scope="")
|
else:
|
||||||
sync_from_root(sess, global_variables) #pylint: disable=E1101
|
global_variables = tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, scope="")
|
||||||
|
sync_from_root(sess, global_variables) #pylint: disable=E1101
|
||||||
|
|
||||||
def train(self, lr, cliprange, obs, returns, masks, actions, values, neglogpacs, states=None):
|
def train(self, lr, cliprange, obs, returns, masks, actions, values, neglogpacs, states=None):
|
||||||
# Here we calculate advantage A(s,a) = R + yV(s') - V(s)
|
# Here we calculate advantage A(s,a) = R + yV(s') - V(s)
|
||||||
|
Reference in New Issue
Block a user