提交 d0ed5e84 编写于 作者: Q QiaoJing 提交者: Jinhui Yuan

fix normlization epsilon check (#1433)



Former-commit-id: 8111c6c82b09d725b2f520744b2e6b3809288c65
上级 18a30e66
......@@ -140,7 +140,7 @@ void NormalizationOp::InferBlobDescsForCudnn(
const BlobDesc* in_blob_desc = GetBlobDesc4BnInOp("in");
const DataType in_data_type = in_blob_desc->data_type();
CHECK(conf.scale() && conf.center()) << "Cudnn batch norm must use scale and center";
CHECK_GT(conf.epsilon(), CUDNN_BN_MIN_EPSILON);
CHECK_GE(conf.epsilon(), CUDNN_BN_MIN_EPSILON);
InferParamBlobDescs(GetBlobDesc4BnInOp, conf, in_blob_desc->shape().At(conf.axis()), in_data_type,
true);
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册