Skip to content

Commit b27b7de

Browse files
Merge pull request #1491 from madeline-underwood/patch-28
Reduce subheading size by 1.
2 parents deda2e0 + 61f60d9 commit b27b7de

File tree

1 file changed

+2
-2
lines changed
  • content/learning-paths/cross-platform/pytorch-digit-classification-arch-training

1 file changed

+2
-2
lines changed

content/learning-paths/cross-platform/pytorch-digit-classification-arch-training/model.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ The total number of trainable parameters for this network is calculated as follo
2525

2626
In total, the network has 102,762 trainable parameters.
2727

28-
# Implementation
28+
## Implementation
2929

3030
To implement the model, supplement the `pytorch-digits.ipynb` notebook with the following statements:
3131

@@ -132,7 +132,7 @@ The output is still a probability distribution over the 10 digit classes (0-9),
132132

133133
Technically, the code will run without errors as long as you provide it with an input image of the correct dimensions, which is 28x28 pixels. The model can accept input, pass it through the layers, and return a prediction - a vector of 10 probabilities. However, the results are not useful until the model is trained.
134134

135-
# What have you learned so far?
135+
## What have you learned so far?
136136

137137
You have successfully defined and initialized a feedforward neural network using PyTorch.
138138

0 commit comments

Comments
 (0)