Skip to content

Commit cc1d420

Browse files
authored
remove set_vital from statefuldataloader (#1537)
1 parent b1e55ec commit cc1d420

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

torchdata/stateful_dataloader/stateful_dataloader.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -376,8 +376,6 @@ def __init__(
376376
# iterator on the next __iter__ call, and this flag is used for those cases.
377377
self._initial_iter_for_state_dict = False
378378

379-
torch.set_vital("Dataloader", "enabled", "True") # type: ignore[attr-defined]
380-
381379
def _get_iterator(self) -> "_StatefulBaseDataLoaderIter":
382380
it: _StatefulBaseDataLoaderIter
383381
if self.num_workers == 0:

0 commit comments

Comments
 (0)