mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Allow non train mode for wds reader to operate w/o sample count, exhaust iterator

Browse Source
This commit is contained in:
Ross Wightman 2024-10-31 15:39:42 -07:00 committed by Ross Wightman
parent f689c850b9
commit f5b58e31a2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
timm/data/readers/reader_wds.py
View File

@ -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)