torchtnt.utils.loggers.InMemoryLogger

class torchtnt.utils.loggers.InMemoryLogger

Simple logger that buffers data in-memory.

Example

from torchtnt.utils.loggers import InMemoryLogger logger = InMemoryLogger() logger.log(“accuracy”, 23.56, 10) logger.close()

__init__() → None

Methods

__init__()
close()	Close log resource, flushing if necessary.
flush()
log(name, data, step)	Log scalar data to the in-memory buffer.
log_dict(payload, step)	Add multiple scalar values.

Attributes

log_buffer

Directly access the buffer.