Update README.md

README.md

@@ -92,10 +92,10 @@ print(output)
 ## Training Procedure 
 
 ```
-deepspeed trainer_sft.py --configs defaults pythia-1b --deepspeed
+deepspeed trainer_sft.py --configs defaults pythia-1-4b-ost --deepspeed
 ```
 
-This model was trained for 1000 iterations.
+This model was trained for 200 iterations. After 200 iterations the accuracy started to drop and loss increasing which is a sign of overfitting.
 
 ### Training Hyperparameters