diff --git a/mindspore/dataset/engine/samplers.py b/mindspore/dataset/engine/samplers.py index 265d20f389..8951a1c4a0 100644 --- 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):