diff options
Diffstat (limited to 'python')
-rw-r--r-- | python/caffe/pycaffe.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/python/caffe/pycaffe.py b/python/caffe/pycaffe.py index 9caa21b8..8a9e1aab 100644 --- a/python/caffe/pycaffe.py +++ b/python/caffe/pycaffe.py @@ -126,7 +126,7 @@ def _Net_forward_all(self, blobs=None, **kwargs): for batch in self._batch(kwargs): outs = self.forward(blobs=blobs, **batch) for out, out_blob in outs.iteritems(): - all_outs[out].extend(out_blob) + all_outs[out].extend(out_blob.copy()) # Package in ndarray. for out in all_outs: all_outs[out] = np.asarray(all_outs[out]) |