lox.loggers.SaveLogger#

class lox.loggers.SaveLogger(path, overwrite=False)#

Bases: Logger[SaveLoggerState]

Logger for saving data to a specified path using JAX’s experimental IO callback.

Undoc-members:

__init__(path, overwrite=False)#

Methods

callback(logger_state, logs)#
init(key)#
Return type:

SaveLoggerState