From 5b0fb713431654f8b9049157e6e4ccfd414074d1 Mon Sep 17 00:00:00 2001 From: yanghaitao Date: Fri, 12 Jun 2020 16:06:03 +0800 Subject: [PATCH] fix SubsetSampler --- mindspore/dataset/engine/samplers.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mindspore/dataset/engine/samplers.py b/mindspore/dataset/engine/samplers.py index b593b193e8..63b8079d33 100644 --- a/mindspore/dataset/engine/samplers.py +++ b/mindspore/dataset/engine/samplers.py @@ -457,6 +457,9 @@ class SubsetRandomSampler(BuiltinSampler): def get_num_samples(self): num_samples = super().get_num_samples() + if num_samples is None: + return len(self.indices) + return min(len(self.indices), num_samples)