diff --git a/python/paddle/nn/functional/input.py b/python/paddle/nn/functional/input.py index fc9c41f2859ee..0794b95c80101 100644 --- a/python/paddle/nn/functional/input.py +++ b/python/paddle/nn/functional/input.py @@ -210,8 +210,8 @@ def embedding(x, weight, padding_idx=None, sparse=False, name=None): padding_idx = -1 if padding_idx is None else padding_idx if padding_idx >= 0 else ( weight.shape[0] + padding_idx) - if padding_idx >= weight.shape[0] or padding_idx <= -weight.shape[0]: - raise ValueError("padding_idx must be within [-{}, {}]".format( + if padding_idx >= weight.shape[0] or padding_idx < -weight.shape[0]: + raise ValueError("padding_idx must be within [-{}, {})".format( weight.shape[0], weight.shape[0])) helper.append_op( diff --git a/python/paddle/nn/layer/common.py b/python/paddle/nn/layer/common.py index 37de833359554..6a3115f166db1 100644 --- a/python/paddle/nn/layer/common.py +++ b/python/paddle/nn/layer/common.py @@ -1673,8 +1673,8 @@ def __init__(self, if self._embedding_dim <= 0: raise ValueError("embedding_dim must be gather than 0") - if self._padding_idx >= num_embeddings or self._padding_idx <= -num_embeddings: - raise ValueError("padding_idx must be within [-{}, {}]".format( + if self._padding_idx >= num_embeddings or self._padding_idx < -num_embeddings: + raise ValueError("padding_idx must be within [-{}, {})".format( num_embeddings, num_embeddings)) self._dtype = self._helper.get_default_dtype()