Skip to content

Commit d78b2ab

Browse files
committed
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 <hargrave@us.ibm.com>
1 parent 2c456a1 commit d78b2ab

File tree

3 files changed

+5
-3
lines changed

3 files changed

+5
-3
lines changed

.jenkins/metadata.json

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,5 +27,8 @@
2727
},
2828
"intermediate_source/torch_compile_tutorial.py": {
2929
"needs": "gpu.nvidia.medium"
30+
},
31+
"beginner_source/introyt/trainingyt.py": {
32+
"needs": "gpu.nvidia.medium"
3033
}
3134
}

.jenkins/validate_tutorials_built.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@
1414
"profiler",
1515
"saving_loading_models",
1616
"introyt/captumyt",
17-
"introyt/trainingyt",
1817
"examples_nn/polynomial_module",
1918
"examples_nn/dynamic_net",
2019
"examples_nn/polynomial_optim",

beginner_source/introyt/trainingyt.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,8 +81,8 @@
8181
validation_set = torchvision.datasets.FashionMNIST('./data', train=False, transform=transform, download=True)
8282

8383
# Create data loaders for our datasets; shuffle for training, not for validation
84-
training_loader = torch.utils.data.DataLoader(training_set, batch_size=4, shuffle=True, num_workers=2)
85-
validation_loader = torch.utils.data.DataLoader(validation_set, batch_size=4, shuffle=False, num_workers=2)
84+
training_loader = torch.utils.data.DataLoader(training_set, batch_size=4, shuffle=True)
85+
validation_loader = torch.utils.data.DataLoader(validation_set, batch_size=4, shuffle=False)
8686

8787
# Class labels
8888
classes = ('T-shirt/top', 'Trouser', 'Pullover', 'Dress', 'Coat',

0 commit comments

Comments
 (0)