fix err msg in pixel shuffle op (#27503)

5 years ago · b7319ef518
parent 273f58a3c5
commit b7319ef518
1 changed files with 2 additions and 2 deletions
--- a/paddle/fluid/operators/pixel_shuffle_op.cc
+++ b/paddle/fluid/operators/pixel_shuffle_op.cc
@ -46,14 +46,14 @@ class PixelShuffleOp : public framework::OperatorWithKernel {
          platform::errors::InvalidArgument(
              "The square of upscale_factor[%u] should divide the "
              "number of channel[%u]",
-              input_dims[1], upscale_factor * upscale_factor));
+              upscale_factor * upscale_factor, input_dims[1]));
    } else {
      PADDLE_ENFORCE_EQ(
          input_dims[3] % (upscale_factor * upscale_factor), 0,
          platform::errors::InvalidArgument(
              "The square of upscale_factor[%u] should divide the "
              "number of channel[%u]",
-              input_dims[3], upscale_factor * upscale_factor));
+              upscale_factor * upscale_factor, input_dims[3]));
    }
    auto output_dims = input_dims;
    output_dims[0] = input_dims[0];