fix subsetrandomsampler

a379c668 · yanghaitao · 0f221403 · a379c668
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

mindspore/dataset/engine/samplers.py mindspore/dataset/engine/samplers.py +1 -1

未找到文件。
--- 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):