def sync_from_root()

in coinrun/main_utils.py [0:0]

• 11 lines of code
• 5 McCabe index (conditional complexity)

def sync_from_root(sess, vars=None):
    if vars is None:
        vars = tf.trainable_variables()

    if Config.SYNC_FROM_ROOT:
        rank = MPI.COMM_WORLD.Get_rank()
        print('sync from root', rank)
        for var in vars:
            if rank == 0:
                MPI.COMM_WORLD.bcast(sess.run(var))
            else:
                sess.run(tf.assign(var, MPI.COMM_WORLD.bcast(None)))