diff options
Diffstat (limited to 'python/caffe/__init__.py')
| -rw-r--r-- | python/caffe/__init__.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/python/caffe/__init__.py b/python/caffe/__init__.py index 1b2da510..6cc44e72 100644 --- a/python/caffe/__init__.py +++ b/python/caffe/__init__.py @@ -1,5 +1,5 @@ from .pycaffe import Net, SGDSolver -from ._caffe import set_mode_cpu, set_mode_gpu, set_device, Layer, get_solver +from ._caffe import set_mode_cpu, set_mode_gpu, set_device, Layer, get_solver, layer_type_list from .proto.caffe_pb2 import TRAIN, TEST from .classifier import Classifier from .detector import Detector |