LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
lbann::adam< TensorDataType > Member List

This is the complete list of members for lbann::adam< TensorDataType >, including all inherited members.

AbsDistMatrixType typedeflbann::adam< TensorDataType >
adam(TensorDataType learning_rate, TensorDataType beta1=0.9, TensorDataType beta2=0.99, TensorDataType eps=1e-8, TensorDataType adamw_weight_decay=0.0)lbann::adam< TensorDataType >
adam(const adam &other)lbann::adam< TensorDataType >
adam()lbann::adam< TensorDataType >inlineprotected
BaseType typedeflbann::adam< TensorDataType >private
callback::perturb_adam classlbann::adam< TensorDataType >friend
clone() constlbann::Cloneable< adam< TensorDataType >, data_type_optimizer< TensorDataType > >inline
get_adamw_weight_decay() const noexceptlbann::adam< TensorDataType >inline
get_beta1() const noexceptlbann::adam< TensorDataType >inline
get_beta2() const noexceptlbann::adam< TensorDataType >inline
get_current_beta1() const noexceptlbann::adam< TensorDataType >inline
get_current_beta2() const noexceptlbann::adam< TensorDataType >inline
get_description() const overridelbann::adam< TensorDataType >
get_eps() const noexceptlbann::adam< TensorDataType >inline
get_moment1() constlbann::adam< TensorDataType >
get_moment1()lbann::adam< TensorDataType >
get_moment2() constlbann::adam< TensorDataType >
get_moment2()lbann::adam< TensorDataType >
get_type() const overridelbann::adam< TensorDataType >inline
m_adamw_weight_decaylbann::adam< TensorDataType >private
m_beta1lbann::adam< TensorDataType >private
m_beta2lbann::adam< TensorDataType >private
m_current_beta1lbann::adam< TensorDataType >private
m_current_beta2lbann::adam< TensorDataType >private
m_epslbann::adam< TensorDataType >private
m_moment1lbann::adam< TensorDataType >private
m_moment2lbann::adam< TensorDataType >private
operator=(const adam &other)lbann::adam< TensorDataType >
OptimizerType typedeflbann::adam< TensorDataType >
serialize(Archive &ar)lbann::adam< TensorDataType >
set_adamw_weight_decay(TensorDataType adamw_weight_decay)lbann::adam< TensorDataType >inline
set_beta1(TensorDataType beta1)lbann::adam< TensorDataType >inline
set_beta2(TensorDataType beta2)lbann::adam< TensorDataType >inline
set_current_beta1(TensorDataType current_beta1)lbann::adam< TensorDataType >inline
set_current_beta2(TensorDataType current_beta2)lbann::adam< TensorDataType >inline
set_eps(TensorDataType eps)lbann::adam< TensorDataType >inline
setup(WeightsType *w=nullptr) overridelbann::adam< TensorDataType >
step_compute(AbsDistMatrixType &values, const AbsDistMatrixType &gradient) overridelbann::adam< TensorDataType >protected
step_compute_cpu(AbsDistMatrixType &values, const AbsDistMatrixType &gradient, const TensorDataType &correction)lbann::adam< TensorDataType >private
WeightsType typedeflbann::adam< TensorDataType >
write_proto(lbann_data::Optimizer &opt) const finallbann::adam< TensorDataType >
~adam()=defaultlbann::adam< TensorDataType >