From 627f98c0b9b739f54e87213a3c06f426180f955b Mon Sep 17 00:00:00 2001 From: carefree0910 Date: Thu, 17 Oct 2024 22:05:11 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=9A=91=EF=B8=8FPatched=20`prepare=5Fdatal?= =?UTF-8?q?oaders`?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- core/learn/schema.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/core/learn/schema.py b/core/learn/schema.py index 807cec3..86b0916 100644 --- 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: