Browse Source

check if num_samples is 0 in sampler

tags/v0.5.0-beta
yanghaitao 5 years ago
parent
commit
9c2a72c55e
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/dataset/engine/samplers.py

+ 2
- 2
mindspore/dataset/engine/samplers.py View File

@@ -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))



Loading…
Cancel
Save