diff options
| author | Gustav Larsson <gustav.m.larsson@gmail.com> | 2015-09-17 20:32:33 -0500 |
|---|---|---|
| committer | Gustav Larsson <gustav.m.larsson@gmail.com> | 2015-09-17 20:33:44 -0500 |
| commit | b4f9add57fa468ab43aa40f0a95badf3e9ace243 (patch) | |
| tree | 9e2094666af3d70efd51a1a14c1cff70cd146cc7 /python | |
| parent | 3d12b5d9cb014d2e0df4db1b10250ef298e549b9 (diff) | |
| download | caffeonacl-b4f9add57fa468ab43aa40f0a95badf3e9ace243.tar.gz caffeonacl-b4f9add57fa468ab43aa40f0a95badf3e9ace243.tar.bz2 caffeonacl-b4f9add57fa468ab43aa40f0a95badf3e9ace243.zip | |
Expose `Snapshot` to pycaffe
- Solver::Snapshot is made public
- It is also added as `snapshot` to pycaffe
Addressing #3077
Diffstat (limited to 'python')
| -rw-r--r-- | python/caffe/_caffe.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/python/caffe/_caffe.cpp b/python/caffe/_caffe.cpp index ccd5776a..6c2ccaa5 100644 --- a/python/caffe/_caffe.cpp +++ b/python/caffe/_caffe.cpp @@ -286,7 +286,8 @@ BOOST_PYTHON_MODULE(_caffe) { .def("solve", static_cast<void (Solver<Dtype>::*)(const char*)>( &Solver<Dtype>::Solve), SolveOverloads()) .def("step", &Solver<Dtype>::Step) - .def("restore", &Solver<Dtype>::Restore); + .def("restore", &Solver<Dtype>::Restore) + .def("snapshot", &Solver<Dtype>::Snapshot); bp::class_<SGDSolver<Dtype>, bp::bases<Solver<Dtype> >, shared_ptr<SGDSolver<Dtype> >, boost::noncopyable>( |