Skip to content

Commit

Permalink
remove unused _pin_memory property in PinMemory (#1413)
Browse files Browse the repository at this point in the history
  • Loading branch information
FightingZhen authored Dec 30, 2024
1 parent 62092dd commit 3f866a8
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion torchdata/nodes/pin_memory.py
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,6 @@ def __init__(
super().__init__()
self.source = source
self.snapshot_frequency = snapshot_frequency
self._pin_memory = torch.cuda.is_available()
if len(pin_memory_device) == 0:
self._pin_memory_device = None
else:
Expand Down

0 comments on commit 3f866a8

Please sign in to comment.