Update README.md

README.md

@@ -41,7 +41,7 @@ Please refer to the [script](https://github.com/huggingface/transformers/tree/ma
 provided by Huggingface. 
 
 
-The model was trained for 400,000 steps with batch size (~2epoches) on 2 NVIDIA Quadro RTX6000 for around 40 hours at the Research Computing Services of Imperial College London. 
+The model was trained for 400,000 steps with batch size 5 (~2epoches) on 2 NVIDIA Quadro RTX6000 for around 40 hours at the Research Computing Services of Imperial College London.