fix datanorm error msg (#30294)

4 years ago · 4656525e24
parent 77051cc9f0
commit 4656525e24
1 changed files with 3 additions and 2 deletions
--- a/paddle/fluid/operators/data_norm_op.cc
+++ b/paddle/fluid/operators/data_norm_op.cc
@ -390,7 +390,7 @@ class DataNormKernel<platform::CPUDeviceContext, T>
      }
      default:
        PADDLE_THROW(platform::errors::InvalidArgument(
-            "Unknown storage order: %d", data_layout));
+            "Unknown storage order: %d, please use NCHW or NHWC", data_layout));
    }
  }
 };
@ -701,7 +701,8 @@ class DataNormGradKernel<platform::CPUDeviceContext, T>
      }
      default:
        PADDLE_THROW(platform::errors::InvalidArgument(
-            "Unknown storage order: %s", data_layout_str));
+            "Unknown storage order: %s, please use NCHW or NHWC",
+            data_layout_str));
    }
  }
 };