From 2e3e31de905816a0032c0861c0e9de9e44300676 Mon Sep 17 00:00:00 2001 From: yanghaitao Date: Thu, 28 May 2020 19:33:29 +0800 Subject: [PATCH] fix subsetrandomsampler --- mindspore/dataset/engine/samplers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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):