mxnet::cpp::LRScheduler Class Reference

lr scheduler interface More...

#include <lr_scheduler.h>

Inheritance diagram for mxnet::cpp::LRScheduler:

Collaboration diagram for mxnet::cpp::LRScheduler:

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_lr

the 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

lr	learning rate from optimizer

Member Data Documentation

float mxnet::cpp::LRScheduler::base_lr_

protected

---

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

• cpp-package/include/mxnet-cpp/lr_scheduler.h