fix SubsetSampler

5 years ago · 5b0fb71343
parent a6a9f88453
commit 5b0fb71343
1 changed files with 3 additions and 0 deletions
--- a/mindspore/dataset/engine/samplers.py
+++ b/mindspore/dataset/engine/samplers.py
@ -457,6 +457,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)