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 

Member Function Documentation

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

get a new learning rate

Implements mxnet::cpp::LRScheduler.

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