chainer.training.extensions.observe_lr

chainer.training.extensions.observe_lr(optimizer_name='main', observation_key='lr')[source]

Returns a trainer extension to record the learning rate.

Parameters:
  • optimizer_name (str) – Name of optimizer whose learning rate is recorded.
  • observation_key (str) – Key of observation to record.
Returns:

The extension function.