pytorch_pfn_extras.training.extensions.observe_lr

pytorch_pfn_extras.training.extensions.observe_lr(optimizer, param_group=0, observation_key='lr')

Returns an extension to record the learning rate.

Parameters

• optimizer (Optimizer) – Optimizer whose learning rate is recorded.

• param_group (int) – Param group of the optimizer to observe

• observation_key (str) – Key of observation to record.

Returns

The extension function.

Return type

Any

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

pytorch_pfn_extras.training.extensions.MicroAverage pytorch_pfn_extras.training.extensions.observe_value