class nbla::AMSBound

template<typename T>
class AMSBound : public nbla::Solver

AMSBound solver.

See also

See the paper linked below for more details. L. Luo, Y. Xiong, Y. Liu and X. Sun. Adaptive Gradient Methods with Dynamic Bound of Learning Rate. https://arxiv.org/abs/1902.09843

Public Functions

inline virtual float learning_rate()

Set learning rate.