Apache SINGA
A distributed deep learning platform .
|
Base learning rate generator. More...
#include <updater.h>
Public Member Functions | |
virtual void | Init (const LRGenProto &proto) |
virtual float | Get (int step) |
Static Public Member Functions | |
static LRGenerator * | Create (const LRGenProto &proto) |
Protected Attributes | |
LRGenProto | proto_ |
Base learning rate generator.
Generate learning rate for a give training step/iteration. There are many different ways to change the learning rate through time/step. Users can inherint this class to implement their own change method.
|
inlinevirtual |
step | training step/iteration. |
Reimplemented in singa::InvTLRGen, singa::InvLRGen, singa::ExpLRGen, singa::LinearLRGen, singa::StepLRGen, and singa::FixedStepLRGen.