!2059 fix SubsetSampler error

Merge pull request !2059 from yanghaitao/yht_fix_subset_samplers
2020-06-12 20:05:06 +08:00 · 2020-06-12 20:05:06 +08:00 · 9282b9473d
parent 1a82132d29 5b0fb71343
commit 9282b9473d
1 changed files with 3 additions and 0 deletions
--- a/mindspore/dataset/engine/samplers.py
+++ b/mindspore/dataset/engine/samplers.py
@ -506,6 +506,9 @@ class SubsetRandomSampler(BuiltinSampler):

    def get_num_samples(self):
        num_samples = super().get_num_samples()
+        if num_samples is None:
+            return len(self.indices)
+
        return min(len(self.indices), num_samples)