bc::nn::Momentum::Optimizer< Tensor > Struct Template Reference

#include <momentum.h>

Inheritance diagram for bc::nn::Momentum::Optimizer< Tensor >:

Collaboration diagram for bc::nn::Momentum::Optimizer< Tensor >:

Public Types
using	value_type = typename Tensor::value_type

Public Member Functions
template<class... Args>
	Optimizer (Args &&... args)
template<class TensorX , class Gradients >
void	update (TensorX &tensor, Gradients &&delta)
void	set_learning_rate (value_type lr)
void	save (Layer_Loader &loader, std::string name) const
void	load (Layer_Loader &loader, std::string name)
Public Member Functions inherited from bc::nn::Optimizer_Base
void	save (Layer_Loader &loader, std::string name) const
void	load (Layer_Loader &loader, std::string name)

Public Attributes
value_type	alpha = .9
value_type	learning_rate = 0.003
Tensor	momentum

Member Typedef Documentation

value_type

template<class Tensor >

using bc::nn::Momentum::Optimizer< Tensor >::value_type = typename Tensor::value_type

Constructor & Destructor Documentation

Optimizer()

template<class Tensor >

template<class... Args>

bc::nn::Momentum::Optimizer< Tensor >::Optimizer ( Args &&...  args )

inline

Member Function Documentation

load()

template<class Tensor >

void bc::nn::Momentum::Optimizer< Tensor >::load ( Layer_Loader &  loader, std::string  name  )

inline

save()

template<class Tensor >

void bc::nn::Momentum::Optimizer< Tensor >::save ( Layer_Loader &  loader, std::string  name  ) const

inline

set_learning_rate()

template<class Tensor >

void bc::nn::Momentum::Optimizer< Tensor >::set_learning_rate ( value_type  lr )

inline

update()

template<class Tensor >

template<class TensorX , class Gradients >

void bc::nn::Momentum::Optimizer< Tensor >::update ( TensorX &  tensor, Gradients &&  delta  )

inline

Member Data Documentation

alpha

template<class Tensor >

value_type bc::nn::Momentum::Optimizer< Tensor >::alpha = .9

learning_rate

template<class Tensor >

value_type bc::nn::Momentum::Optimizer< Tensor >::learning_rate = 0.003

momentum

template<class Tensor >

Tensor bc::nn::Momentum::Optimizer< Tensor >::momentum

---

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

• blackcat/neural_networks/optimzers/momentum.h