From a29c2f7a0ff2ff4278a2e498f0b686b5d5cb88cd Mon Sep 17 00:00:00 2001 From: Alex Lee Date: Sat, 14 Nov 2015 12:49:05 -0800 Subject: [PATCH] Fix outs and diffs being overwritten in forward_backward_all. --- python/caffe/pycaffe.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/caffe/pycaffe.py b/python/caffe/pycaffe.py index 7bd4f411b6a..31dc702f647 100644 --- a/python/caffe/pycaffe.py +++ b/python/caffe/pycaffe.py @@ -216,9 +216,9 @@ def _Net_forward_backward_all(self, blobs=None, diffs=None, **kwargs): batch_blobs = self.forward(blobs=blobs, **fb) batch_diffs = self.backward(diffs=diffs, **bb) for out, out_blobs in batch_blobs.iteritems(): - all_outs[out].extend(out_blobs) + all_outs[out].extend(out_blobs.copy()) for diff, out_diffs in batch_diffs.iteritems(): - all_diffs[diff].extend(out_diffs) + all_diffs[diff].extend(out_diffs.copy()) # Package in ndarray. for out, diff in zip(all_outs, all_diffs): all_outs[out] = np.asarray(all_outs[out])