Search code examples
mxnet

mxnet: is there's a way to monitor, in the call back, the learning rate value?


For debugging purpose, I would like to monitor in my fit callback func the learning rate value to make sure my mx.lr_scheduler.MultiFactorScheduler does the job as expected.

Unfortunately the learning rate does not seem the be accessible in the Params. Is there a way do access the actual used lr for the current batch?

Many thanks !


Solution

  • One possible solution may be creating a subclass from the current optimizer you use. Then override update method to log the current learning rate for every updates.