Method SetMomentumGenerator

include/singa/model/optimizer.h:187–189

The argument function returns the momentum value given the current running step (i.e., iterations/mini-batches).

  /// The argument function returns the momentum value given the current running
  /// step (i.e., iterations/mini-batches).
  void SetMomentumGenerator(std::function<float(int)> func) {
    momentum_generator_ = func;
  }

 private:
  std::unordered_map<string, Tensor> history_gradient_;