From 2b77b254f212a6c54e9a013dca546fd3e2356ce9 Mon Sep 17 00:00:00 2001 From: Doyeon Kim <84639487+haughty-yeon@users.noreply.github.com> Date: Thu, 23 Nov 2023 10:58:54 +0900 Subject: [PATCH] distributed sampler length modififed --- src/torchio/data/queue.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/torchio/data/queue.py b/src/torchio/data/queue.py index c0090881..a2034ee3 100644 --- a/src/torchio/data/queue.py +++ b/src/torchio/data/queue.py @@ -258,7 +258,7 @@ def subjects_iterable(self): @property def num_subjects(self) -> int: if self.subject_sampler is not None: - return len(self.subject_sampler) + return self.subject_sampler.__len__() return len(self.subjects_dataset) @property