fix subsetrandomsampler

5 years ago · 2e3e31de90
parent ad279e90fd
commit 2e3e31de90
1 changed files with 1 additions and 1 deletions
--- a/mindspore/dataset/engine/samplers.py
+++ b/mindspore/dataset/engine/samplers.py
@ -483,7 +483,7 @@ class SubsetRandomSampler(BuiltinSampler):


    def get_dataset_size(self):
-        return len(indices)
+        return len(self.indices)


 class WeightedRandomSampler(BuiltinSampler):