mxnet: есть ли способ контролировать, при обратном вызове, значение скорости обучения?

В целях отладки я хотел бы отслеживать в моей функции обратного вызова fit fit значение скорости обучения, чтобы убедиться, что мой mx.lr_scheduler.MultiFactorScheduler выполняет свою работу, как ожидалось.

К сожалению, скорость обучения не представляется доступной в Params. Есть ли способ получить доступ к фактически используемому lr для текущей партии?

Большое спасибо!

1 ответ

Решение

Одним из возможных решений может быть создание подкласса из текущего оптимизатора, который вы используете. Затем переопределите метод обновления, чтобы записать текущую скорость обучения для каждого обновления.

Другие вопросы по тегам