diff --git a/README.md b/README.md
index cd9464e..57cbab8 100644
--- a/README.md
+++ b/README.md
@@ -5,9 +5,9 @@ tags:
 - video-classification
 ---
 
-# VideoMAE (base-sized model, pre-trained only for 1600 epochs) 
+# VideoMAE (base-sized model, pre-trained only) 
 
-VideoMAE model pre-trained on Kinetics-400 in a self-supervised way. It was introduced in the paper [VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training](https://arxiv.org/abs/2203.12602) by Tong et al. and first released in [this repository](https://github.com/MCG-NJU/VideoMAE). 
+VideoMAE model pre-trained on Kinetics-400 for 1600 epochs in a self-supervised way. It was introduced in the paper [VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training](https://arxiv.org/abs/2203.12602) by Tong et al. and first released in [this repository](https://github.com/MCG-NJU/VideoMAE). 
 
 Disclaimer: The team releasing VideoMAE did not write a model card for this model so this model card has been written by the Hugging Face team.