Fix SwitchOrderLayer grad bugs by reshape output.grad

baae8447 · wanghaoshuang · a6c53fc2 · baae8447
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

paddle/gserver/layers/SwitchOrderLayer.cpp paddle/gserver/layers/SwitchOrderLayer.cpp +1 -0

未找到文件。
--- 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