Python Multi-GPU

[platform/upstream/caffeonacl.git] / python / caffe / pycaffe.py

diff --git a/python/caffe/pycaffe.py b/python/caffe/pycaffe.py
index 5bae18d..1880381 100644 (file)
--- a/python/caffe/pycaffe.py
+++ b/python/caffe/pycaffe.py
@@ -11,7 +11,7 @@ except:
 import numpy as np
 
 from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, \
-        RMSPropSolver, AdaDeltaSolver, AdamSolver
+        RMSPropSolver, AdaDeltaSolver, AdamSolver, NCCL, Timer
 import caffe.io
 
 import six