From 59ffab537c7913dcbc555bed0cca87f9a2770924 Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Thu, 17 Mar 2022 14:42:40 -0700
Subject: [PATCH] Fix mistake in wds sample slicing

---
 timm/data/parsers/parser_wds.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/timm/data/parsers/parser_wds.py b/timm/data/parsers/parser_wds.py
index 7011d967..02ce2794 100644
--- a/timm/data/parsers/parser_wds.py
+++ b/timm/data/parsers/parser_wds.py
@@ -285,7 +285,7 @@ class ParserWebdataset(Parser):
 
     def _split_by_node_and_worker(self, src):
         if self.global_num_workers > 1:
-            for s in islice(src, self.global_worker_id, self.global_num_workers):
+            for s in islice(src, self.global_worker_id, None, self.global_num_workers):
                 yield s
         else:
             for s in src: