Class Momentum

include/core/optimizer.h:279–286

Helper class for Momentum */

/** Helper class for Momentum */
class Momentum : public Optimizer {
public:
    Momentum(float _lr = 1e-4, float _weight_decay = 0, float _momentum = 0.999,
             const LRSchedule &_schedule = "linear") :
            Optimizer("Momentum", 1, _lr, _weight_decay, _schedule) {
        momentum = _momentum;
    }
};

/** Helper class for AdaGrad */
class AdaGrad : public Optimizer {