We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 58029f2 commit b9f3978Copy full SHA for b9f3978
utils/utils.py
@@ -92,8 +92,7 @@ def init_voc_dataset(config: dict):
92
testloader = torch.utils.data.DataLoader(testset,
93
batch_size=config['batch_size'],
94
shuffle=False,
95
- num_workers=config['num_workers'],
96
- pin_memory=True)
+ num_workers=config['num_workers'])
97
98
return trainset, trainloader, testset, testloader
99
0 commit comments