Merge pull request #5514 from pkuyym/fix-5513

Bug fix for ScaleSubRegionLayer
8 years ago · 9876bb0952
parent 5a38195688 4cd859c578
commit 9876bb0952
1 changed files with 1 additions and 1 deletions
--- a/paddle/gserver/layers/ScaleSubRegionLayer.cpp
+++ b/paddle/gserver/layers/ScaleSubRegionLayer.cpp
@ -49,7 +49,7 @@ void ScaleSubRegionLayer::forward(PassType passType) {
  shape_ = TensorShape({batchSize, channelsNum_, imgH_, imgW_});
  resetOutput(batchSize, imgV->getWidth());
-  auto out = getOutput();
+  auto& out = getOutput();
  out.setFrameHeight(imgH_);
  out.setFrameWidth(imgW_);