diff --git a/README.md b/README.md
index da65856..203ab87 100644
--- a/README.md
+++ b/README.md
@@ -5,7 +5,7 @@ tags:
 - video-classification
 ---
 
-# VideoMAE (base-sized model, pre-trained only) 
+# VideoMAE (base-sized model, pre-trained only for 1600 epochs) 
 
 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).