match existing python formatting
authorEvan Shelhamer <shelhamer@imaginarynumber.net>
Wed, 9 Apr 2014 21:02:40 +0000 (14:02 -0700)
committerEvan Shelhamer <shelhamer@imaginarynumber.net>
Wed, 14 May 2014 01:10:41 +0000 (18:10 -0700)
python/caffe/pycaffe.py

index 05187e9..6b7b00e 100644 (file)
@@ -14,30 +14,35 @@ from ._caffe import Net, SGDSolver
 
 @property
 def _Net_blobs(self):
-    """
-    An OrderedDict (bottom to top, i.e., input to output) of network
-    blobs indexed by name
-    """
-    return OrderedDict([(bl.name, bl) for bl in self._blobs])
+  """
+  An OrderedDict (bottom to top, i.e., input to output) of network
+  blobs indexed by name
+  """
+  return OrderedDict([(bl.name, bl) for bl in self._blobs])
 
 Net.blobs = _Net_blobs
 
+
 @property
 def _Net_params(self):
-    """
-    An OrderedDict (bottom to top, i.e., input to output) of network
-    parameters indexed by name; each is a list of multiple blobs (e.g.,
-    weights and biases)
-    """
-    return OrderedDict([(lr.name, lr.blobs) for lr in self.layers
-                                            if len(lr.blobs) > 0])
+  """
+  An OrderedDict (bottom to top, i.e., input to output) of network
+  parameters indexed by name; each is a list of multiple blobs (e.g.,
+  weights and biases)
+  """
+  return OrderedDict([(lr.name, lr.blobs) for lr in self.layers
+                                          if len(lr.blobs) > 0])
 
 Net.params = _Net_params
 
 def _Net_set_input_arrays(self, data, labels):
-    if labels.ndim == 1:
-        labels = np.ascontiguousarray(labels[:, np.newaxis, np.newaxis,
-                                             np.newaxis])
-    return self._set_input_arrays(data, labels)
+  """
+  Set input arrays of the in-memory MemoryDataLayer.
+  (Note: this is only for networks declared with the memory data layer.)
+  """
+  if labels.ndim == 1:
+    labels = np.ascontiguousarray(labels[:, np.newaxis, np.newaxis,
+                                         np.newaxis])
+  return self._set_input_arrays(data, labels)
 
 Net.set_input_arrays = _Net_set_input_arrays