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 5043bf5 commit 216f783Copy full SHA for 216f783
beginner_source/introyt/introyt1_tutorial.py
@@ -358,7 +358,7 @@ def num_flat_features(self, x):
358
#
359
360
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,
361
- shuffle=True, num_workers=0)
+ shuffle=True, num_workers=2)
362
363
364
##########################################################################
@@ -436,12 +436,12 @@ def imshow(img):
436
trainset = torchvision.datasets.CIFAR10(root='./data', train=True,
437
download=True, transform=transform)
438
439
440
441
testset = torchvision.datasets.CIFAR10(root='./data', train=False,
442
443
testloader = torch.utils.data.DataLoader(testset, batch_size=4,
444
- shuffle=False, num_workers=0)
+ shuffle=False, num_workers=2)
445
446
classes = ('plane', 'car', 'bird', 'cat',
447
'deer', 'dog', 'frog', 'horse', 'ship', 'truck')
0 commit comments