memorax.loggers.TensorBoardLogger

Contents

memorax.loggers.TensorBoardLogger#

class memorax.loggers.TensorBoardLogger[source]#

Bases: object

__init__(directory='tensorboard', num_seeds=1, **kwargs)[source]#

log(data, step, **kwargs)[source]#

Return type:

Parameters:

data (Any)
step (int)

finish()[source]#

Return type:: None