From a379c668f58af70b2ceaf951e415feb841777b90 Mon Sep 17 00:00:00 2001 From: yanghaitao Date: Thu, 28 May 2020 19:38:16 +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):