ConvNet  1.0
A GPU-based C++ implementation of Convolutional Neural Nets
 All Classes Namespaces Functions Variables
multigpu_convnet.h
1 #ifndef MULTIGPU_CONVNET_H_
2 #define MULTIGPU_CONVNET_H_
3 #include "convnet.h"
4 
5 class MultiGPUConvNet : public ConvNet {
6  public:
7  MultiGPUConvNet(const string& model_file);
8 
9  protected:
10  virtual void Fprop(bool train);
11  virtual void Bprop(bool update_weights);
12 };
13 #endif
virtual void Bprop(bool update_weights)
Backpropagate through the network and update weights.
Definition: multigpu_convnet.cc:49
Definition: multigpu_convnet.h:5
virtual void Fprop(bool train)
Forward propagate through the network.
Definition: multigpu_convnet.cc:6
A Convolutional Net Model.
Definition: convnet.h:14