Apache Singa
A General Distributed Deep Learning Library
|
Basic Updater class just forward all the method function call to the wrapped Optimizer. More...
#include <updater.h>
Public Member Functions | |
Updater (Optimizer *opt) | |
virtual void | Setup (const OptimizerConf &conf) |
Forward Setup() to Optimizer. | |
virtual void | Register (const string &name, const ParamSpec &specs) |
Forward Register() to Optimizer. | |
virtual void | Apply (int step, const string &name, Tensor &grad, Tensor &value) |
Forward Apply() to Optimizer. | |
Optimizer * | GetOptimizer () |
Updater (const Updater &)=delete | |
void | operator= (const Updater &)=delete |
Protected Attributes | |
Optimizer * | opt_ |