restored vituals in solver.hpp

deniskorobicyn · Sep 1, 2014 · 8a9c268 · 8a9c268
1 parent 9d10569
commit 8a9c268
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/include/caffe/solver.hpp b/include/caffe/solver.hpp
@@ -66,11 +66,11 @@ class SGDSolver : public Solver<Dtype> {
       : Solver<Dtype>(param_file) {}
 
  protected:
-  void PreSolve();
+  virtual void PreSolve();
   Dtype GetLearningRate();
   virtual void ComputeUpdateValue();
-  void SnapshotSolverState(SolverState * state);
-  void RestoreSolverState(const SolverState& state);
+  virtual void SnapshotSolverState(SolverState * state);
+  virtual void RestoreSolverState(const SolverState& state);
   // history maintains the historical momentum data.
   // update maintains update related data and is not needed in snapshots.
   vector<shared_ptr<Blob<Dtype> > > history_, update_;