diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index 072679696b1b2565643be9a5597696bd10cff275..004dcf7382a72b2a8ace728c8eed36cc1ce40b03 100644 --- a/python/paddle/fluid/layers/nn.py +++ b/python/paddle/fluid/layers/nn.py @@ -3029,10 +3029,6 @@ def hsigmoid(input, label, num_classes=2, param_attr=None, bias_attr=None): dim = input.shape[1] if num_classes < 2: raise ValueError("num_classes must be lager or equal than 2.") - if input.shape[0] != label.shape[1]: - raise ValueError( - "input's 1-st dimension and label's 2-nd dimension must be equal, they both equal to batch size." - ) weights = helper.create_parameter( attr=helper.param_attr, shape=[num_classes - 1, dim],