未验证 提交 b7319ef5 编写于 作者: R ruri 提交者: GitHub

fix err msg in pixel shuffle op (#27503)

上级 273f58a3
...@@ -46,14 +46,14 @@ class PixelShuffleOp : public framework::OperatorWithKernel { ...@@ -46,14 +46,14 @@ class PixelShuffleOp : public framework::OperatorWithKernel {
platform::errors::InvalidArgument( platform::errors::InvalidArgument(
"The square of upscale_factor[%u] should divide the " "The square of upscale_factor[%u] should divide the "
"number of channel[%u]", "number of channel[%u]",
input_dims[1], upscale_factor * upscale_factor)); upscale_factor * upscale_factor, input_dims[1]));
} else { } else {
PADDLE_ENFORCE_EQ( PADDLE_ENFORCE_EQ(
input_dims[3] % (upscale_factor * upscale_factor), 0, input_dims[3] % (upscale_factor * upscale_factor), 0,
platform::errors::InvalidArgument( platform::errors::InvalidArgument(
"The square of upscale_factor[%u] should divide the " "The square of upscale_factor[%u] should divide the "
"number of channel[%u]", "number of channel[%u]",
input_dims[3], upscale_factor * upscale_factor)); upscale_factor * upscale_factor, input_dims[3]));
} }
auto output_dims = input_dims; auto output_dims = input_dims;
output_dims[0] = input_dims[0]; output_dims[0] = input_dims[0];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册