diff --git a/.jenkins/validate_tutorials_built.py b/.jenkins/validate_tutorials_built.py index 5d627b70398..7811fa0509f 100644 --- a/.jenkins/validate_tutorials_built.py +++ b/.jenkins/validate_tutorials_built.py @@ -14,7 +14,6 @@ "profiler", "saving_loading_models", "introyt/captumyt", - "introyt/trainingyt", "examples_nn/polynomial_module", "examples_nn/dynamic_net", "examples_nn/polynomial_optim", diff --git a/beginner_source/introyt/trainingyt.py b/beginner_source/introyt/trainingyt.py index 6b6bf8a8b80..929e06c1b57 100644 --- a/beginner_source/introyt/trainingyt.py +++ b/beginner_source/introyt/trainingyt.py @@ -81,8 +81,8 @@ validation_set = torchvision.datasets.FashionMNIST('./data', train=False, transform=transform, download=True) # Create data loaders for our datasets; shuffle for training, not for validation -training_loader = torch.utils.data.DataLoader(training_set, batch_size=4, shuffle=True, num_workers=2) -validation_loader = torch.utils.data.DataLoader(validation_set, batch_size=4, shuffle=False, num_workers=2) +training_loader = torch.utils.data.DataLoader(training_set, batch_size=4, shuffle=True) +validation_loader = torch.utils.data.DataLoader(validation_set, batch_size=4, shuffle=False) # Class labels classes = ('T-shirt/top', 'Trouser', 'Pullover', 'Dress', 'Coat',