Class AdaGrad

include/core/optimizer.h:289–296

Helper class for AdaGrad */

/** Helper class for AdaGrad */
class AdaGrad : public Optimizer {
public:
    AdaGrad(float _lr = 1e-4, float _weight_decay = 0, float _epsilon = 1e-10,
            const LRSchedule &_schedule = "linear") :
            Optimizer("AdaGrad", 1, _lr, _weight_decay, _schedule) {
        epsilon = _epsilon;
    }
};

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