LRMonitor

class composer.callbacks.lr_monitor.LRMonitor[source]

Bases: composer.core.callback.Callback

Logs the learning rate.

Learning rates for each optimizer is logged on each batch under the lr-{OPTIMIZER_NAME}/group{GROUP_NUMBER} key.

batch_end(state: State, logger: Logger)[source]

Called on the BATCH_END event.

Parameters
  • state (State) – The global state.

  • logger (Logger) – The logger.