mxnet
Public Member Functions | List of all members
mxnet::cpp::FactorScheduler Class Reference

#include <lr_scheduler.h>

Inheritance diagram for mxnet::cpp::FactorScheduler:
Inheritance graph
Collaboration diagram for mxnet::cpp::FactorScheduler:
Collaboration graph

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_
 

Constructor & Destructor Documentation

mxnet::cpp::FactorScheduler::FactorScheduler ( int  step,
float  factor = 1,
float  stop_factor_lr = 1e-8 
)
inlineexplicit

Member Function Documentation

float mxnet::cpp::FactorScheduler::GetLR ( unsigned  num_update)
inlineoverridevirtual

get a new learning rate

Implements mxnet::cpp::LRScheduler.


The documentation for this class was generated from the following file: