ConvNet  1.0
A GPU-based C++ implementation of Convolutional Neural Nets
 All Classes Namespaces Functions Variables
grad_check.h
1 #ifndef GRAD_CHECK_H_
2 #define GRAD_CHECK_H_
3 
4 #include "convnet.h"
5 
6 class GradChecker : public ConvNet {
7  public:
8  GradChecker(const string& model_file);
9  void Run();
10  void Run(Matrix& w, Matrix& dLbydw_analytical, float epsilon, const vector<float>& analytical_error);
11 
12  protected:
13  virtual void GetLoss(vector<float>& error);
14 };
15 
16 #endif
Definition: grad_check.h:6
virtual void GetLoss(vector< float > &error)
Computes the loss function (to be displayed).
Definition: grad_check.cc:8
A Convolutional Net Model.
Definition: convnet.h:14
A GPU matrix class.
Definition: matrix.h:11