python/caffe/__init__.py

   1 from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver
   2 from ._caffe import init_log, log, set_mode_cpu, set_mode_gpu, set_device, Layer, get_solver, layer_type_list, set_random_seed
   3 from ._caffe import __version__
   4 from .proto.caffe_pb2 import TRAIN, TEST
   5 from .classifier import Classifier
   6 from .detector import Detector
   7 from . import io
   8 from .net_spec import layers, params, NetSpec, to_proto