!1609 fix subset random sampler error

Merge pull request !1609 from yanghaitao/yht_subsetrandomsampler
5 years ago · 83b74e4693
parent 21da86b393 2e3e31de90
commit 83b74e4693
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):