Example #1
0
 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)
Example #2
0
    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)