diff --git a/baselines/trpo_mpi/run_mujoco.py b/baselines/trpo_mpi/run_mujoco.py index f25aee0..220bb91 100644 --- a/baselines/trpo_mpi/run_mujoco.py +++ b/baselines/trpo_mpi/run_mujoco.py @@ -12,7 +12,10 @@ def train(env_id, num_timesteps, seed): sess.__enter__() rank = MPI.COMM_WORLD.Get_rank() - if rank != 0: + if rank == 0: + logger.configure() + else: + logger.configure(format_strs=[]) logger.set_level(logger.DISABLED) workerseed = seed + 10000 * MPI.COMM_WORLD.Get_rank() def policy_fn(name, ob_space, ac_space): @@ -25,9 +28,9 @@ def train(env_id, num_timesteps, seed): def main(): args = mujoco_arg_parser().parse_args() - logger.configure() train(args.env, num_timesteps=args.num_timesteps, seed=args.seed) if __name__ == '__main__': main() +