diff options
| author | Jonathan L Long <jonlong@cs.berkeley.edu> | 2014-04-25 14:09:02 -0700 |
|---|---|---|
| committer | Jonathan L Long <jonlong@cs.berkeley.edu> | 2014-05-02 13:25:51 -0700 |
| commit | 634a382bce186e63671d37b4d8e40939a74b6373 (patch) | |
| tree | 802aef6d881d801c23699e33f959c4576cacfb8d /python | |
| parent | ce30b678f62da135d350d78097978ee8756c99f8 (diff) | |
| download | caffe-634a382bce186e63671d37b4d8e40939a74b6373.tar.gz caffe-634a382bce186e63671d37b4d8e40939a74b6373.tar.bz2 caffe-634a382bce186e63671d37b4d8e40939a74b6373.zip | |
pycaffe: let boost pass shared_ptr<CaffeNet>
Diffstat (limited to 'python')
| -rw-r--r-- | python/caffe/_caffe.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/python/caffe/_caffe.cpp b/python/caffe/_caffe.cpp index a443f897..1a44debf 100644 --- 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) |