Update README.md

README.md

@@ -33,7 +33,7 @@ It was trained using a [synthetically created, annotated dataset](https://huggin
 
 ### Training
 
-Training was conducted on a 10 epoch fine-tuning approach, however this repository contains the results of the second epoch, since it has the best accuracy:
+Training was conducted on a 10 epoch fine-tuning approach, however this repository contains the results of the fourth epoch, since it has the best accuracy:
 
 | epoch | accuracy          | loss               |
 |-------|-------------------|--------------------|
@@ -50,7 +50,7 @@ Training was conducted on a 10 epoch fine-tuning approach, however this reposito
 
         
 
-In relation to the dataset, the model demonstrates that it can effectively learn to distinguish between the two classes claim and premise. However, the rapid onset of overfitting after epoch 2 suggests that the dataset is imbalanced and noisy. Further work should enable the model to be trained on more robust data to ensure better evaluation results.
+In relation to the dataset, the model demonstrates that it can effectively learn to distinguish between the two classes claim and premise. However, the rapid onset of overfitting after epoch 4 suggests that the dataset is imbalanced and noisy. Further work should enable the model to be trained on more robust data to ensure better evaluation results.
 
 ### Text Classification Tags