From 936633e514a606194f92734cab138d90c49a78bb Mon Sep 17 00:00:00 2001 From: mcuiaws Date: Thu, 24 Oct 2024 23:48:22 -0700 Subject: [PATCH] parallel_loader: fix AttributeError (#8314) (#8315) --- torch_xla/distributed/parallel_loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_xla/distributed/parallel_loader.py b/torch_xla/distributed/parallel_loader.py index 7053361f795..22046df75e2 100644 --- a/torch_xla/distributed/parallel_loader.py +++ b/torch_xla/distributed/parallel_loader.py @@ -131,7 +131,7 @@ def per_device_loader(self, device): return PerDeviceLoader(self, torch.device(device)) def per_device_samples(self): - return len(self._loader) // len(self._devices) + return len(self._cpu_loader) // len(self._devices) def next_item(self, device): dqueue = self._queues[device]