Update README.md

README.md

@@ -17,11 +17,11 @@ The model was trained with the [MNIST digits classification dataset](https://ker
 - Size: 1,256,080 parameters
 - Dataset: [tf.keras.datasets.mnist](https://storage.googleapis.com/tensorflow/tf-keras-datasets/mnist.npz)
 - Language: English
-- Number of Training Steps: 10
+- Number of Training Epochs: 10
 - Batch size: 32
 - Optimizer: Adam
 - Learning Rate: 0.001
-- GPU: T4
+- Hardware: Tesla T4
 - This repository has the source [code used](https://github.com/Nkluge-correa/teeny-tiny_castle/blob/master/ML%20Intro%20Course/19_quick_AI_app.ipynb) to train this model.
 
 # Cite as 🤗