Class Adam

include/core/optimizer.h:310–319

Helper class for Adam */

/** Helper class for Adam */
class Adam : public Optimizer {
public:
    Adam(float _lr = 1e-4, float _weight_decay = 0, float _beta1 = 0.999, float _beta2 = 0.99999, float _epsilon = 1e-8,
         const LRSchedule &_schedule = "linear") :
            Optimizer("Adam", 2, _lr, _weight_decay, _schedule) {
        beta1 = _beta1;
        beta2 = _beta2;
        epsilon = _epsilon;
    }
};

} // namespace graphvite