[pycaffe] expose SGDSolver::Step as SGDSolver.step

BVLC · Oct 7, 2014 · 5ac0faf · 5ac0faf
1 parent e57bb92
commit 5ac0faf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/python/caffe/_caffe.cpp b/python/caffe/_caffe.cpp
@@ -195,7 +195,8 @@ BOOST_PYTHON_MODULE(_caffe) {
       .add_property("test_nets", &PySGDSolver::test_nets)
       .add_property("iter",      &PySGDSolver::iter)
       .def("solve",              &PySGDSolver::Solve)
-      .def("solve",              &PySGDSolver::SolveResume);
+      .def("solve",              &PySGDSolver::SolveResume)
+      .def("step",               &PySGDSolver::Step);
 
   bp::class_<vector<shared_ptr<PyNet> > >("NetVec")
       .def(bp::vector_indexing_suite<vector<shared_ptr<PyNet> >, true>());

diff --git a/python/caffe/_caffe.hpp b/python/caffe/_caffe.hpp
@@ -176,6 +176,7 @@ class PySGDSolver {
   int iter() { return solver_->iter(); }
   void Solve() { return solver_->Solve(); }
   void SolveResume(const string& resume_file);
+  void Step(int niter) { solver_->Step(niter); }
 
  protected:
   shared_ptr<PyNet> net_;