fix_loader_state (#1457)

pytorch · Feb 26, 2025 · c8cdedf · c8cdedf
1 parent d155220
commit c8cdedf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchdata/nodes/loader.py b/torchdata/nodes/loader.py
@@ -123,7 +123,7 @@ def next(self):
             self._cached_state_dict = None
         else:
             item = next(self.root)
-        self._num_yielded += 1
+            self._num_yielded += 1
         return item
 
     def get_state(self) -> Dict[str, Any]: