memorax.loggers.TensorBoardLogger

memorax.loggers.TensorBoardLogger#

class memorax.loggers.TensorBoardLogger[source]#

Bases: BaseLogger[TensorBoardLoggerState]

TensorBoardLogger(log_dir: str = ‘tensorboard’)

log_dir: str = 'tensorboard'#
init(cfg)[source]#
Return type:

TensorBoardLoggerState

Parameters:

cfg (dict)

log(state, data, step)[source]#
Return type:

TensorBoardLoggerState

Parameters:

  • state (TensorBoardLoggerState)

  • data (Any)

  • step (int)

emit(state)[source]#
Return type:

TensorBoardLoggerState

Parameters:

state (TensorBoardLoggerState)

finish(state)[source]#
Return type:

None

Parameters:

state (TensorBoardLoggerState)

classmethod __class_getitem__(params)#

Parameterizes a generic class.

At least, parameterizing a generic class is the main thing this method does. For example, for some generic class Foo, this is called when we do Foo[int] - there, with cls=Foo and params=int.

However, note that this method is also called when defining generic classes in the first place with class Foo(Generic[T]): ….

__init__(log_dir='tensorboard')#
Parameters:

log_dir (str)

Return type:

None

replace(**updates)#

Returns a new object replacing the specified fields with new values.

Contents