From: Evan Shelhamer Date: Wed, 9 Apr 2014 21:02:40 +0000 (-0700) Subject: match existing python formatting X-Git-Tag: submit/tizen/20180823.020014~692^2~1^2~27 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=51f276e6b20f18c41079fab01177988491825536;p=platform%2Fupstream%2Fcaffeonacl.git match existing python formatting --- diff --git a/python/caffe/pycaffe.py b/python/caffe/pycaffe.py index 05187e9..6b7b00e 100644 --- a/python/caffe/pycaffe.py +++ b/python/caffe/pycaffe.py @@ -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