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

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

AbsDistMatrixType typedeflbann::sgd< TensorDataType >
BaseType typedeflbann::sgd< TensorDataType >private
cereal::access classlbann::sgd< TensorDataType >friend
clone() constlbann::Cloneable< sgd< TensorDataType >, data_type_optimizer< TensorDataType > >inline
get_description() const overridelbann::sgd< TensorDataType >
get_momentum() const noexceptlbann::sgd< TensorDataType >inline
get_type() const overridelbann::sgd< TensorDataType >inline
get_velocity() constlbann::sgd< TensorDataType >
get_velocity()lbann::sgd< TensorDataType >
m_momentumlbann::sgd< TensorDataType >private
m_nesterovlbann::sgd< TensorDataType >private
m_velocitylbann::sgd< TensorDataType >private
momentum_step_cpu(AbsDistMatrixType &values, const AbsDistMatrixType &gradient)lbann::sgd< TensorDataType >private
operator=(const sgd &other)lbann::sgd< TensorDataType >
OptimizerType typedeflbann::sgd< TensorDataType >
serialize(ArchiveT &ar)lbann::sgd< TensorDataType >
set_momentum(TensorDataType momentum)lbann::sgd< TensorDataType >inline
set_nesterov(bool nesterov)lbann::sgd< TensorDataType >inline
setup(WeightsType *w=nullptr) overridelbann::sgd< TensorDataType >
sgd(TensorDataType learning_rate, TensorDataType momentum=0, bool nesterov=false)lbann::sgd< TensorDataType >
sgd(const sgd &other)lbann::sgd< TensorDataType >
sgd()lbann::sgd< TensorDataType >inlineprotected
step_compute(AbsDistMatrixType &values, const AbsDistMatrixType &gradient) overridelbann::sgd< TensorDataType >protected
using_nesterov() const noexceptlbann::sgd< TensorDataType >inline
WeightsType typedeflbann::sgd< TensorDataType >
write_proto(lbann_data::Optimizer &opt) const finallbann::sgd< TensorDataType >
~sgd() override=defaultlbann::sgd< TensorDataType >