save the info

[ChenyangRan]

Hi, the info includes a flag which represents succ or fail to grasp.
How can I save the info by tensorboard?
I don't see the env.step and save the info.

[Trinkle23897]

Here it is: I add a log_fn in both collector and trainer. This hook receives each step's env info.

class Logger(object):
    def __init__(self, writer):
        self.cnt = 0
        self.writer = writer

    def log(self, info):
        self.writer.add_scalar('key', info['key'], global_step=self.cnt)
        self.cnt += 1

and in trainer (or collector, if you don't use trainer), add an argument log_fn=logger.log.

[ChenyangRan]

Here it is: I add a log_fn in both collector and trainer. This hook receives each step's env info.

class Logger(object):
    def __init__(self, writer):
        self.cnt = 0
        self.writer = writer

    def log(self, info):
        self.writer.add_scalar('key', info['key'], global_step=self.cnt)
        self.cnt += 1

and in trainer (or collector, if you don't use trainer), add an argument log_fn=logger.log.

Thanks, I'll try it.