Method SetLearningRateGenerator

include/singa/model/optimizer.h:81–83

The argument is a function that returns the learning rate given the current step (i.e., curren running iteration).

  /// The argument is a function that returns the learning rate given the
  /// current step (i.e., curren running iteration).
  void SetLearningRateGenerator(function<float(int)> func) {
    learning_rate_generator_ = func;
  }
  float GetLearningRate(int step) {
    if (learning_rate_generator_)
      return learning_rate_generator_(step);