Apache SINGA
A distributed deep learning platform .
|
#include <updater.h>
Public Member Functions | |
virtual void | Init (const UpdaterProto &proto) |
virtual void | Update (int step, Param *param, float grad_scale)=0 |
Static Public Member Functions | |
static Updater * | Create (const UpdaterProto &proto) |
Protected Attributes | |
UpdaterProto | proto_ |
LRGenerator * | lr_gen_ |
float | weight_decay_ |
float | momentum_ |