Update README.md

README.md

@@ -67,7 +67,7 @@ cos(a, b)
 * Training was conducted in two stages:
 1. In the first stage, the model was trained on Wikipedia texts (4 million texts) for three epochs.
    <img src="https://github.com/FractalGPT/ModelEmbedderDistilation/blob/main/DistilSBERT/Train/1_st_en.JPG?raw=true" width=700 />
-3. In the second stage, training was conducted on Wikipedia, a dialog dataset, and NLI for one epoch.
+3. In the second stage, training was conducted on Wikipedia and dialog dataset for one epoch.
    <img src="https://github.com/FractalGPT/ModelEmbedderDistilation/blob/main/DistilSBERT/Train/2_st_en.JPG?raw=true" width=700 />
 
 ## Full Model Architecture