Inheritance diagram for singa::RMSProp:
Collaboration diagram for singa::RMSProp:
Public Member Functions | |
| void | Setup (const OptimizerConf &conf) |
| Setup the meta fields of the optimizer. | |
| void | Apply (int epoch, float lr, const string &name, Tensor &grad, Tensor &value, int step=-1) override |
| Apply the updating algorithm. | |
 Public Member Functions inherited from singa::Optimizer | |
| void | Setup (const string &str) |
| Setup the optimzier using configurations from serialized string (for binding languages). More... | |
| virtual void | Register (const string &name, const ParamSpec &specs) |
| Register the parameter, e.g., create Constraint and Regularizers. More... | |
| virtual void | ApplyRegularizerConstraint (int epoch, const string &name, const Tensor &value, Tensor &grad, int step=-1) |
| void | Apply (int epoch, const string &name, Tensor &grad, Tensor &value, int step=-1) |
| Apply the updating algorithm if the gradient is not empty. More... | |
| void | SetLearningRateGenerator (function< float(int)> func) |
| The argument is a function that returns the learning rate given the current step (i.e., curren running iteration). More... | |
| float | GetLearningRate (int step) |
Additional Inherited Members | |
| Protected Attributes inherited from singa::Optimizer | |
| function< float(int)> | learning_rate_generator_ |
| std::unordered_map< std::string, float > | learning_rate_multplier_ |
| std::unordered_map< std::string, Constraint * > | constraints_ |
| std::unordered_map< std::string, Regularizer * > | regularizers_ |
| Constraint * | constraint_ = nullptr |
| Regularizer * | regularizer_ = nullptr |
| OptimizerConf | conf_ |
The documentation for this class was generated from the following file:
- /home/moaz/incubator-singa/include/singa/model/optimizer.h
