Update datasets-and-training.md

Author: madeline-underwood

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

@@ -70,9 +70,9 @@ The certifi Python package provides the Mozilla root certificates, which are ess
 Make sure to replace 'x' with the version number of Python that you have installed.
 {{% /notice %}}
 
-After running the code, you will see output similar to Figure 1:
+After running the code, you will see output similar to Figure 5:
 
-![image alt-text#center](Figures/01.png "Figure 1. Output 1".)
+![image alt-text#center](Figures/01.png "Figure 5. Output".)
 
 ## Train the Model