提交 9c2a72c5 编写于 作者: Y yanghaitao

check if num_samples is 0 in sampler

上级 1973594b
......@@ -340,7 +340,7 @@ class RandomSampler(BuiltinSampler):
raise ValueError("replacement should be a boolean value, but got replacement={}".format(replacement))
if num_samples is not None:
if num_samples < 0:
if num_samples <= 0:
raise ValueError("num_samples should be a positive integer "
"value, but got num_samples={}".format(num_samples))
......@@ -490,7 +490,7 @@ class WeightedRandomSampler(BuiltinSampler):
weights = [weights]
if num_samples is not None:
if num_samples < 0:
if num_samples <= 0:
raise ValueError("num_samples should be a positive integer "
"value, but got num_samples={}".format(num_samples))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册