|
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.
|
|
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) |
|
|
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