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

497 Views Asked by At

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 !

1

There are 1 best solutions below

3
On BEST ANSWER

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.