diff options
Diffstat (limited to 'python/caffe/_caffe.cpp')
| -rw-r--r-- | python/caffe/_caffe.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/python/caffe/_caffe.cpp b/python/caffe/_caffe.cpp index e1ae3ec7..020a5bee 100644 --- a/python/caffe/_caffe.cpp +++ b/python/caffe/_caffe.cpp @@ -200,6 +200,8 @@ BOOST_PYTHON_MODULE(_caffe) { bp::def("set_mode_gpu", &set_mode_gpu); bp::def("set_device", &Caffe::SetDevice); + bp::def("layer_type_list", &LayerRegistry<Dtype>::LayerTypeList); + bp::class_<Net<Dtype>, shared_ptr<Net<Dtype> >, boost::noncopyable >("Net", bp::no_init) .def("__init__", bp::make_constructor(&Net_Init)) |