dnn::opt_rmsprop Class Reference

RMSprop optimizer class. More...

#include <dnn_opt.h>

Inheritance diagram for dnn::opt_rmsprop:

Public Member Functions
	opt_rmsprop (DNN_Dtype s, DNN_Dtype l=0.0, DNN_Dtype a=0.0, const DNN_Dtype b=0.9, DNN_Dtype e=1e-8)
	RMSprop constructor. More...
	~opt_rmsprop ()
void	apply (arma::Cube< DNN_Dtype > &W, arma::Mat< DNN_Dtype > &B, const arma::Cube< DNN_Dtype > &Wgrad, const arma::Mat< DNN_Dtype > &Bgrad)
	Apply the optimizer to the layer parameters. More...
std::string	get_algorithm (void)
	Get the optimizer algorithm information. More...
Public Member Functions inherited from dnn::opt
	opt ()
	~opt ()
void	set_learn_rate_alg (LR_ALG alg, DNN_Dtype a=0.0, DNN_Dtype b=10.0)
	Set learning rate algorithm. More...
void	update_learn_rate (void)
	Update learning rate. More...
DNN_Dtype	get_learn_rate (void)
	Get the learning rate. More...

Private Attributes
arma::Cube< DNN_Dtype >	v
arma::Mat< DNN_Dtype >	vB
DNN_Dtype	beta
DNN_Dtype	eps

Additional Inherited Members
Protected Attributes inherited from dnn::opt
std::string	alg
DNN_Dtype	lr
	Learning rate. More...
DNN_Dtype	reg_lambda
	Regularisation parameter lambda. More...
DNN_Dtype	reg_alpha
	Elastic net mix parameter - 0=ridge (L2) .. 1=LASSO (L1) More...
LR_ALG	lr_alg
	Learning rate schedule algorithm. More...
DNN_Dtype	lr_0
	Init value for lr. More...
DNN_Dtype	lr_a
	Internal parameter a. More...
DNN_Dtype	lr_b
	Internal parameter b. More...
arma::uword	it
	Iteration counter. More...

Detailed Description

RMSprop optimizer class.

RMSprop algorithm, see https://arxiv.org/pdf/1609.04747.pdf

Definition at line 731 of file dnn_opt.h.

Constructor & Destructor Documentation

opt_rmsprop()

dnn::opt_rmsprop::opt_rmsprop ( DNN_Dtype  s, DNN_Dtype  l = 0.0, DNN_Dtype  a = 0.0, const DNN_Dtype  b = 0.9, DNN_Dtype  e = 1e-8  )

inline

RMSprop constructor.

Parameters

[in]	s	Step size - learning rate
[in]	l	Regularisation param lambda
[in]	a	Regularisation param alpha
[in]	b	beta parameter
[in]	e	eps parameter

Definition at line 747 of file dnn_opt.h.

~opt_rmsprop()

dnn::opt_rmsprop::~opt_rmsprop ( )

inline

Definition at line 758 of file dnn_opt.h.

Member Function Documentation

apply()

void dnn::opt_rmsprop::apply ( arma::Cube< DNN_Dtype > &  W, arma::Mat< DNN_Dtype > &  B, const arma::Cube< DNN_Dtype > &  Wgrad, const arma::Mat< DNN_Dtype > &  Bgrad  )

inlinevirtual

Apply the optimizer to the layer parameters.

Parameters

[in,out]	W,B	Learnable parameters
[in]	Wgrad,Bgrad	Gradient of the learnable parameters

Implements dnn::opt.

Definition at line 765 of file dnn_opt.h.

get_algorithm()

std::string dnn::opt_rmsprop::get_algorithm ( void  )

inlinevirtual

Get the optimizer algorithm information.

Returns

Algorithm information string

Reimplemented from dnn::opt.

Definition at line 801 of file dnn_opt.h.

Member Data Documentation

beta

DNN_Dtype dnn::opt_rmsprop::beta

private

Definition at line 736 of file dnn_opt.h.

eps

DNN_Dtype dnn::opt_rmsprop::eps

private

Definition at line 737 of file dnn_opt.h.

v

arma::Cube<DNN_Dtype> dnn::opt_rmsprop::v

private

Definition at line 734 of file dnn_opt.h.

vB

arma::Mat<DNN_Dtype> dnn::opt_rmsprop::vB

private

Definition at line 735 of file dnn_opt.h.

---

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

• dnn_opt.h