File: __init__.py

package info (click to toggle)
caffe-contrib 1.0.0~rc4-1
  • links: PTS, VCS
  • area: contrib
  • in suites: stretch
  • size: 16,276 kB
  • ctags: 9,281
  • sloc: cpp: 60,050; python: 5,649; makefile: 606; sh: 559
file content (8 lines) | stat: -rw-r--r-- 561 bytes parent folder | download | duplicates (2) 
1
2
3
4
5
6
7
8
 
from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver, NCCL, Timer
from ._caffe import init_log, log, set_mode_cpu, set_mode_gpu, set_device, Layer, get_solver, layer_type_list, set_random_seed, solver_count, set_solver_count, solver_rank, set_solver_rank, set_multiprocess, Layer, get_solver
from ._caffe import __version__
from .proto.caffe_pb2 import TRAIN, TEST
from .classifier import Classifier
from .detector import Detector
from . import io
from .net_spec import layers, params, NetSpec, to_proto