From f5b58e31a2d40027daf6b84a98b750d32c6e4ec9 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 31 Oct 2024 15:39:42 -0700 Subject: [PATCH] Allow non train mode for wds reader to operate w/o sample count, exhaust iterator --- timm/data/readers/reader_wds.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/data/readers/reader_wds.py b/timm/data/readers/reader_wds.py index b0fd71d9..7b014992 100644 --- a/timm/data/readers/reader_wds.py +++ b/timm/data/readers/reader_wds.py @@ -304,8 +304,8 @@ class ReaderWds(Reader): self.num_samples = num_samples else: self.num_samples = self.split_info.num_samples - if not self.num_samples: - raise RuntimeError(f'Invalid split definition, num_samples not specified.') + if is_training and not self.num_samples: + raise RuntimeError(f'Invalid split definition, num_samples not specified in train mode.') self.remap_class = False if class_map: self.class_to_idx = load_class_map(class_map)