Class AdamOptimizer

Inheritance Relationships

Base Type

Class Documentation

class AdamOptimizer : public GradientBasedOptimizer

Public Functions

inline AdamOptimizer(UINT trainable_parameter_count, double learning_rate = 0.001, double beta1 = 0.9, double beta2 = 0.999, double epsilon = 1e-8)
inline virtual ~AdamOptimizer()
inline virtual void apply_gradient(std::vector<double> *parameter, const std::vector<double> &gradient) override