diff options
| author | Jeff Donahue <jeff.donahue@gmail.com> | 2015-08-20 11:04:05 -0700 |
|---|---|---|
| committer | Jeff Donahue <jeff.donahue@gmail.com> | 2015-08-20 11:04:05 -0700 |
| commit | aa2a6f55b9e50b29d607aaee0fae19bd085d6565 (patch) | |
| tree | 60e2d7214f66fe265586e0766ab588c1cac07db7 /python/caffe/_caffe.cpp | |
| parent | a7cf704f8ecf48b1ad2ad6b0a7b5ee9e787bbe17 (diff) | |
| parent | 51b172ce2fcd7f63aa7830389af54d353f53a3bc (diff) | |
| download | caffeonacl-aa2a6f55b9e50b29d607aaee0fae19bd085d6565.tar.gz caffeonacl-aa2a6f55b9e50b29d607aaee0fae19bd085d6565.tar.bz2 caffeonacl-aa2a6f55b9e50b29d607aaee0fae19bd085d6565.zip | |
Merge pull request #2930 from lukeyeager/pycaffe-layer_type_list
Expose LayerFactory::LayerTypeList in pycaffe
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)) |