def __init__(self, env, rank, log_dir, **kwargs): self.log_dir = log_dir self.rank = rank self.render_gui = kwargs.get('render', False) self.render_rank = kwargs.get('render_rank', 0) if log_dir is not None: log_dir = os.path.join(log_dir, str(rank)) Monitor.__init__(self, env, log_dir)
def __init__(self, env, rank, log_dir, **kwargs): self.log_dir = log_dir self.rank = rank global_render = kwargs.get("render", False) render_rank = kwargs.get("render_rank", 0) self.render_me = False if global_render and self.rank == render_rank: self.render_me = True if log_dir is not None: log_dir = os.path.join(log_dir, str(rank)) Monitor.__init__(self, env, log_dir)