Class AdamOptimizer¶
Defined in File optimizer.hpp
Inheritance Relationships¶
Base Type¶
public GradientBasedOptimizer(Class GradientBasedOptimizer)
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¶
-
inline AdamOptimizer(UINT trainable_parameter_count, double learning_rate = 0.001, double beta1 = 0.9, double beta2 = 0.999, double epsilon = 1e-8)¶