Apache SINGA
A distributed deep learning platform .
|
Public Member Functions | |
virtual void | Init (const UpdaterProto &proto) |
virtual void | Update (int step, shared_ptr< Param > param, float grad_scale=1.0f) |
Public Member Functions inherited from singa::Updater | |
float | GetLearningRate (int step) |
Protected Attributes | |
float | base_lr_ |
float | momentum_ |
float | weight_decay_ |
Protected Attributes inherited from singa::Updater | |
UpdaterProto | proto_ |