ceras
yet another deep learning engine
Public Types | Public Member Functions | Public Attributes | List of all members
ceras::sgd< Loss, T > Struct Template Reference

#include <optimizer.hpp>

Inheritance diagram for ceras::sgd< Loss, T >:

Public Types

typedef tensor< T > tensor_type
 

Public Member Functions

 sgd (Loss &loss, std::size_t batch_size, T learning_rate=1.0e-1, T momentum=0.0, T decay=0.0, bool nesterov=false) noexcept
 
void forward ()
 

Public Attributes

Loss & loss_
 
learning_rate_
 
momentum_
 
decay_
 
bool nesterov_
 
unsigned long iterations_
 

Member Typedef Documentation

◆ tensor_type

template<typename Loss , typename T >
typedef tensor< T > ceras::sgd< Loss, T >::tensor_type

Constructor & Destructor Documentation

◆ sgd()

template<typename Loss , typename T >
ceras::sgd< Loss, T >::sgd ( Loss &  loss,
std::size_t  batch_size,
learning_rate = 1.0e-1,
momentum = 0.0,
decay = 0.0,
bool  nesterov = false 
)
inlinenoexcept

Member Function Documentation

◆ forward()

template<typename Loss , typename T >
void ceras::sgd< Loss, T >::forward ( )
inline

Member Data Documentation

◆ decay_

template<typename Loss , typename T >
T ceras::sgd< Loss, T >::decay_

◆ iterations_

template<typename Loss , typename T >
unsigned long ceras::sgd< Loss, T >::iterations_

◆ learning_rate_

template<typename Loss , typename T >
T ceras::sgd< Loss, T >::learning_rate_

◆ loss_

template<typename Loss , typename T >
Loss& ceras::sgd< Loss, T >::loss_

◆ momentum_

template<typename Loss , typename T >
T ceras::sgd< Loss, T >::momentum_

◆ nesterov_

template<typename Loss , typename T >
bool ceras::sgd< Loss, T >::nesterov_

The documentation for this struct was generated from the following file: