MiniDNN
Optimizer.h
1 #ifndef OPTIMIZER_H_
2 #define OPTIMIZER_H_
3 
4 #include <Eigen/Core>
5 #include "Config.h"
6 
7 namespace MiniDNN {
8 
9 
13 
19 class Optimizer
20 {
21 protected:
22  typedef Eigen::Matrix<Scalar, Eigen::Dynamic, 1> Vector;
23  typedef Vector::ConstAlignedMapType ConstAlignedMapVec;
24  typedef Vector::AlignedMapType AlignedMapVec;
25 
26 public:
27  virtual ~Optimizer() {}
28 
32  virtual void reset() {};
33 
45  virtual void update(ConstAlignedMapVec& dvec, AlignedMapVec& vec) = 0;
46 };
47 
48 
49 } // namespace MiniDNN
50 
51 
52 #endif /* OPTIMIZER_H_ */
virtual void reset()
Definition: Optimizer.h:32
virtual void update(ConstAlignedMapVec &dvec, AlignedMapVec &vec)=0