From 9c2a72c55e602a1c89ea24d3108a4a021ff751b3 Mon Sep 17 00:00:00 2001 From: yanghaitao Date: Sat, 6 Jun 2020 11:40:09 +0800 Subject: [PATCH] check if num_samples is 0 in sampler --- mindspore/dataset/engine/samplers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/dataset/engine/samplers.py b/mindspore/dataset/engine/samplers.py index 3ae917dd41..b593b193e8 100644 --- a/mindspore/dataset/engine/samplers.py +++ b/mindspore/dataset/engine/samplers.py @@ -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))