Skip to content

Commit b9f3978

Browse files
committed
utils: Remove pin memory
1 parent 58029f2 commit b9f3978

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

utils/utils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,8 +92,7 @@ def init_voc_dataset(config: dict):
9292
testloader = torch.utils.data.DataLoader(testset,
9393
batch_size=config['batch_size'],
9494
shuffle=False,
95-
num_workers=config['num_workers'],
96-
pin_memory=True)
95+
num_workers=config['num_workers'])
9796

9897
return trainset, trainloader, testset, testloader
9998

0 commit comments

Comments
 (0)