Fix SwitchOrderLayer grad bugs by reshape output.grad

8 years ago · baae8447ac
parent a6c53fc2fc
commit baae8447ac
1 changed files with 1 additions and 0 deletions
--- a/paddle/gserver/layers/SwitchOrderLayer.cpp
+++ b/paddle/gserver/layers/SwitchOrderLayer.cpp
@ -81,6 +81,7 @@ void SwitchOrderLayer::forward(PassType passType) {
  resetOutput(outDims_[0], outDims_[1] * outDims_[2] * outDims_[3]);
  if (heightAxis_.size() > 0) {
    getOutputValue()->reshape(reshapeHeight_, reshapeWidth_);
+    getOutputGrad()->reshape(reshapeHeight_, reshapeWidth_);
  }

  // switch NCHW to NHWC