Skip to content

Commit 216f783

Browse files
committed
Updated
1 parent 5043bf5 commit 216f783

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

beginner_source/introyt/introyt1_tutorial.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -358,7 +358,7 @@ def num_flat_features(self, x):
358358
#
359359

360360
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,
361-
shuffle=True, num_workers=0)
361+
shuffle=True, num_workers=2)
362362

363363

364364
##########################################################################
@@ -436,12 +436,12 @@ def imshow(img):
436436
trainset = torchvision.datasets.CIFAR10(root='./data', train=True,
437437
download=True, transform=transform)
438438
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,
439-
shuffle=True, num_workers=0)
439+
shuffle=True, num_workers=2)
440440

441441
testset = torchvision.datasets.CIFAR10(root='./data', train=False,
442442
download=True, transform=transform)
443443
testloader = torch.utils.data.DataLoader(testset, batch_size=4,
444-
shuffle=False, num_workers=0)
444+
shuffle=False, num_workers=2)
445445

446446
classes = ('plane', 'car', 'bird', 'cat',
447447
'deer', 'dog', 'frog', 'horse', 'ship', 'truck')

0 commit comments

Comments
 (0)