From b9f194d884930a1dc7b35b1cf0208100f42c9f12 Mon Sep 17 00:00:00 2001 From: Ramanish Singh Date: Wed, 5 Feb 2025 12:40:05 -0800 Subject: [PATCH] remove commented lines --- torchdata/stateful_dataloader/sampler.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/torchdata/stateful_dataloader/sampler.py b/torchdata/stateful_dataloader/sampler.py index bddef097e..5fc232973 100644 --- a/torchdata/stateful_dataloader/sampler.py +++ b/torchdata/stateful_dataloader/sampler.py @@ -113,11 +113,6 @@ def load_state_dict(self, state_dict: Dict[str, Any]) -> None: # We skip x samples if underlying sampler is not stateful for _ in range(self.samples_yielded): next(self.sampler_iter) - # elif self.samples_yielded > 0: - # print("no fast forward, reset") - # # don't re-create sampler_iter unless necessary, we may already have one from init - # self.sampler_iter = iter(self.sampler) - # self.samples_yielded = 0 class BatchSampler(torch.utils.data.sampler.BatchSampler):