Class SGD

include/core/optimizer.h:272–276

Helper class for SGD */

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

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