Update README.md

README.md

@@ -13,7 +13,7 @@ Entire dataset is synthetic. Synthetic data is used because there is little to n
 This by means a no replacement to a Doctor or professional therapist. If you are in stress or going through a tough time, please seek professional help or talk to a friend/family member.
 
 **Training:**
-Entire dataset was trained on Azure 4 x A100 80GB. Training took 29 hours. DeepSpeed codebase was used for training purpose. 
+Entire dataset was trained on Azure 4 x A100 80GB. For 3 epoch, training took 29 hours. DeepSpeed codebase was used for training purpose. 
 
 **Example Prompt:**
 ```