🚑️Patched prepare_dataloaders

carefree0910 · Oct 17, 2024 · 627f98c · 627f98c
1 parent ada6474
commit 627f98c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/core/learn/schema.py b/core/learn/schema.py
@@ -412,7 +412,8 @@ def _iter(self: DataLoader) -> Iterator[tensor_dict_type]:
             base = d.base_dataloader
             base.async_prefetch = loader.async_prefetch
             base.async_prefetch_factor = loader.async_prefetch_factor
-            base._get_iterator = loader._get_iterator
+            if base.async_prefetch:
+                base._get_iterator = loader._get_iterator
             td = type(d)
             iter_prepared = getattr(td, "_iter_prepared_", False)
             if not iter_prepared: