gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix logger error for MPI

Browse Source 
Can't run logger.configure() if rank != 0
This commit is contained in:
20chase
2018-01-25 22:09:00 +08:00
committed by GitHub
parent b5be53dc92
commit ac2ea4f31f
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
baselines/trpo_mpi/run_mujoco.py
View File

@@ -19,7 +19,10 @@ def train(env_id, num_timesteps, seed):
sess.__enter__() sess.__enter__()
rank = MPI.COMM_WORLD.Get_rank() rank = MPI.COMM_WORLD.Get_rank()
if rank != 0: if rank == 0:
logger.configure()
else:
logger.configure(format_strs=[])
logger.set_level(logger.DISABLED) logger.set_level(logger.DISABLED)
workerseed = seed + 10000 * MPI.COMM_WORLD.Get_rank() workerseed = seed + 10000 * MPI.COMM_WORLD.Get_rank()
set_global_seeds(workerseed) set_global_seeds(workerseed)
@@ -43,7 +46,6 @@ def main():
parser.add_argument('--seed', help='RNG seed', type=int, default=0) parser.add_argument('--seed', help='RNG seed', type=int, default=0)
parser.add_argument('--num-timesteps', type=int, default=int(1e6)) parser.add_argument('--num-timesteps', type=int, default=int(1e6))
args = parser.parse_args() args = parser.parse_args()
logger.configure()
train(args.env, num_timesteps=args.num_timesteps, seed=args.seed) train(args.env, num_timesteps=args.num_timesteps, seed=args.seed)