Skip to content

Commit a21516b

Browse files
committed
Avoid setting pin_memory in tests
1 parent 3f0087b commit a21516b

3 files changed

Lines changed: 0 additions & 3 deletions

File tree

source/tests/pt/model/test_saveload_dpa1.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,6 @@ def setUp(self) -> None:
6969
batch_size=None,
7070
num_workers=0, # setting to 0 diverges the behavior of its iterator; should be >=1
7171
drop_last=False,
72-
pin_memory=True,
7372
)
7473

7574
def cycle_iterator(iterable):

source/tests/pt/model/test_saveload_se_e2_a.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,6 @@ def setUp(self) -> None:
6969
batch_size=None,
7070
num_workers=0, # setting to 0 diverges the behavior of its iterator; should be >=1
7171
drop_last=False,
72-
pin_memory=True,
7372
)
7473

7574
def cycle_iterator(iterable):

source/tests/pt/test_sampler.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,6 @@ def test_sampler_debug_info(self) -> None:
6262
batch_size=None,
6363
num_workers=0, # setting to 0 diverges the behavior of its iterator; should be >=1
6464
drop_last=False,
65-
pin_memory=True,
6665
)
6766
with torch.device("cpu"):
6867
batch_data = next(iter(dataloader))

0 commit comments

Comments
 (0)