mxnet
Public Member Functions | Protected Attributes | List of all members
mxnet::cpp::LRScheduler Class Referenceabstract

lr scheduler interface More...

#include <lr_scheduler.h>

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

Public Member Functions

 LRScheduler (float base_lr=0.01)
 constructor More...
 
void SetLR (const float lr)
 set base lr More...
 
virtual float GetLR (unsigned num_update)=0
 get a new learning rate More...
 
virtual ~LRScheduler ()
 destructor More...
 

Protected Attributes

float base_lr_
 

Detailed Description

lr scheduler interface

Constructor & Destructor Documentation

mxnet::cpp::LRScheduler::LRScheduler ( float  base_lr = 0.01)
inlineexplicit

constructor

Parameters
base_lrthe initial learning rate.
virtual mxnet::cpp::LRScheduler::~LRScheduler ( )
inlinevirtual

destructor

Member Function Documentation

virtual float mxnet::cpp::LRScheduler::GetLR ( unsigned  num_update)
pure virtual

get a new learning rate

Implemented in mxnet::cpp::FactorScheduler.

void mxnet::cpp::LRScheduler::SetLR ( const float  lr)
inline

set base lr

Parameters
lrlearning rate from optimizer

Member Data Documentation

float mxnet::cpp::LRScheduler::base_lr_
protected

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