Update README.md

README.md

@@ -13,7 +13,7 @@ tags:
 - vit
 - doclaynet
 ---
-# ViT Model for Document Layout Classification
+# Vision Transformer(ViT) for Document Classification(DocLayNet)
 
 This model is a fine-tuned Vision Transformer (ViT) for document layout classification based on the DocLayNet dataset.
 
@@ -38,7 +38,9 @@ DocLayNet is a comprehensive dataset for document layout analysis, containing va
 
 ## Training procedure
 
-The training was made with following hyperparameters:
+Trained for 20 epochs on a single gpu for ~10 mins.
+
+The training hyperparameters:
 
 ```python
 {