pycaffe: let boost pass shared_ptr<CaffeNet>

diff --git a/python/caffe/_caffe.cpp b/python/caffe/_caffe.cpp
index a443f89..1a44deb 100644 (file)
--- a/python/caffe/_caffe.cpp
+++ b/python/caffe/_caffe.cpp
@@ -317,7 +317,7 @@ class CaffeSGDSolver {
 
 // The boost python module definition.
 BOOST_PYTHON_MODULE(_caffe) {
-  boost::python::class_<CaffeNet>(
+  boost::python::class_<CaffeNet, shared_ptr<CaffeNet> >(
       "Net", boost::python::init<string, string>())
       .def(boost::python::init<string>())
       .def("Forward",          &CaffeNet::Forward)