add norm_op for ssd(cross channel norm)

7 years ago · d8b13dee5e
parent b1025cf50a
commit d8b13dee5e
1 changed files with 3 additions and 0 deletions
--- a/paddle/operators/norm_op.cc
+++ b/paddle/operators/norm_op.cc
@ -68,6 +68,9 @@ class NormOp : public framework::OperatorWithKernel {
    PADDLE_ENFORCE(ctx->HasInput("X"),
                   "Input(X) of NormOp"
                   "should not be null.");
+    PADDLE_ENFORCE(ctx->HasInput("Scale"),
+                   "Input(Scale) of NormOp"
+                   "should not be null.");
    PADDLE_ENFORCE(ctx->HasOutput("Out"),
                   "Output(Out) of NormOp should not be null.");
    auto in_x_dims = ctx->GetInputDim("X");