test=develop

7 years ago · 99d1446a8b
parent db06568e69
commit 99d1446a8b
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@ -4412,7 +4412,7 @@ def hsigmoid(input,
    out = helper.create_variable_for_type_inference(dtype)
    pre_out = helper.create_variable_for_type_inference(dtype)
    dim = input.shape[1]
-    if ((num_classes < 2) or (num_classes is None)) and (not is_costum):
+    if ((num_classes is None) or (num_classes < 2)) and (not is_costum):
        raise ValueError(
            "num_classes must not be less than 2 with default tree")