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.

This extension is triggered each epoch by default. To change this, use the trigger argument with the Trainer.extend() method.