solver: added snapshotting ability

[jacinto-ai/caffe-jacinto.git] / src / caffe / optimization / solver.hpp

diff --git a/src/caffe/optimization/solver.hpp b/src/caffe/optimization/solver.hpp
index a5ea61262a815c8c9cd7283e1d5e99c4a028806d..98c872dc21d573f1f3b70c9f8ab4212ac88e80f3 100644 (file)
--- a/src/caffe/optimization/solver.hpp
+++ b/src/caffe/optimization/solver.hpp
@@ -27,7 +27,7 @@ class Solver {
   // that stores the learned net. You should implement the SnapshotSolverState()
   // function that produces a SolverState protocol buffer that needs to be
   // written to disk together with the learned net.
-  void Snapshot(bool is_final = false);
+  void Snapshot();
   virtual void SnapshotSolverState(SolverState* state) = 0;
   // The Restore function implements how one should restore the solver to a
   // previously snapshotted state. You should implement the RestoreSolverState()