tiny_dnn
1.0.0
A header only, dependency-free deep learning framework in C++11
|
SGD with momentum. More...
#include <optimizer.h>
Public Member Functions | |
void | update (const vec_t &dW, vec_t &W, bool parallelize) |
![]() | |
void | reset () override |
![]() | |
optimizer (const optimizer &)=default | |
optimizer (optimizer &&)=default | |
optimizer & | operator= (const optimizer &)=default |
optimizer & | operator= (optimizer &&)=default |
Public Attributes | |
float_t | alpha |
float_t | lambda |
float_t | mu |
Additional Inherited Members | |
![]() | |
vec_t & | get (const vec_t &key) |
![]() | |
std::unordered_map< const vec_t *, vec_t > | E_ [N] |
SGD with momentum.
B T Polyak, Some methods of speeding up the convergence of iteration methods USSR Computational Mathematics and Mathematical Physics, 4(5):1-17, 1964.