Class RMSprop

include/core/optimizer.h:299–307

Helper class for RMSprop */

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

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