From 3819a52946e4144597b3cc8f8a260fc615531105 Mon Sep 17 00:00:00 2001 From: BJ Hargrave Date: Fri, 17 Feb 2023 10:05:01 -0500 Subject: [PATCH] Include introyt/trainingyt tutorial in the build The tutorial was not being built and published on the website. So fixes such as 7d8cb43 were not then visible on the website since old output files are still present. Signed-off-by: BJ Hargrave --- .jenkins/validate_tutorials_built.py | 1 - beginner_source/introyt/trainingyt.py | 4 ++-- 2 files changed, 2 insertions(+), 3 deletions(-) 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',