memorax.loggers.CheckpointLogger

Contents

memorax.loggers.CheckpointLogger#

class memorax.loggers.CheckpointLogger[source]#

Bases: object

__init__(directory='checkpoints', max_to_keep=None, **kwargs)[source]#

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

Parameters:

data (Any)
step (int)
train_state (Any | None)

finish()[source]#

Return type:: None