Update README.md

README.md

@@ -61,7 +61,7 @@ It achieves the following results on the evaluation set:
 - (Full length is 200,000+, We will upload full trained Model soon)
 
 **GPU**: One A100 GPU
-**Train time**: About 20 hours for 3 epochs
+**Train time**: About 24 hours for 3 epochs
 
 **Test time**: About 8 hours for test dataset.
 
@@ -84,7 +84,7 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 390
-- num_epochs: 3
+- **num_epochs: 3 (takes about 24 hours)**
 
 ### Training results