mxnet
|
#include <lr_scheduler.h>
Public Member Functions | |
FactorScheduler (int step, float factor=1, float stop_factor_lr=1e-8) | |
float | GetLR (unsigned num_update) override |
get a new learning rate More... | |
Public Member Functions inherited from mxnet::cpp::LRScheduler | |
LRScheduler (float base_lr=0.01) | |
constructor More... | |
void | SetLR (const float lr) |
set base lr More... | |
virtual | ~LRScheduler () |
destructor More... | |
Additional Inherited Members | |
Protected Attributes inherited from mxnet::cpp::LRScheduler | |
float | base_lr_ |
|
inlineexplicit |
|
inlineoverridevirtual |
get a new learning rate
Implements mxnet::cpp::LRScheduler.