fix indentation error on traininglayers.py

Apex-94 · Aug 10, 2016 · 94b2a53 · 94b2a53
1 parent b30ee0a
commit 94b2a53
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 12 deletions.
diff --git a/caffe-colorization b/caffe-colorization
diff --git a/resources/caffe_traininglayers.py b/resources/caffe_traininglayers.py
@@ -144,16 +144,15 @@ def backward(self, top, propagate_down, bottom):
             bottom[i].diff[...] = np.zeros_like(bottom[i].data)
 
 class ClassRebalanceMultLayer(caffe.Layer):
-'''
-INPUTS
-    bottom[0]   NxMxXxY     feature map
-    bottom[1]   Nx1xXxY     boost coefficients
-OUTPUTS
-    top[0]      NxMxXxY     on forward, gets copied from bottom[0]
-FUNCTIONALITY
-    On forward pass, top[0] passes bottom[0]
-    On backward pass, bottom[0] gets boosted by bottom[1]
-    through pointwise multiplication (with singleton expansion) '''
+    ''' INPUTS
+        bottom[0]   NxMxXxY     feature map
+        bottom[1]   Nx1xXxY     boost coefficients
+    OUTPUTS
+        top[0]      NxMxXxY     on forward, gets copied from bottom[0]
+    FUNCTIONALITY
+        On forward pass, top[0] passes bottom[0]
+        On backward pass, bottom[0] gets boosted by bottom[1]
+        through pointwise multiplication (with singleton expansion) '''
     def setup(self, bottom, top):
         # check input pair
         if len(bottom)==0:
@@ -250,7 +249,6 @@ def __init__(self,NN,sigma,km_filepath='',cc=-1):
         self.alreadyUsed = False
 
     def encode_points_mtx_nd(self,pts_nd,axis=1,returnSparse=False,sameBlock=True):
-        t = rz.Timer();
         pts_flt = flatten_nd_array(pts_nd,axis=axis)
         P = pts_flt.shape[0]
         if(sameBlock and self.alreadyUsed):

diff --git a/resources/caffe_traininglayers.pyc b/resources/caffe_traininglayers.pyc
diff --git a/train/solver.prototxt b/train/solver.prototxt
@@ -6,7 +6,7 @@ base_lr: 3.16e-5
 lr_policy: "step"
 gamma: 0.316
 stepsize: 215000
-display: 100
+display: 1
 max_iter: 500000
 momentum: 0.9
 momentum2: 0.99