Update README.md

README.md

@@ -17,9 +17,9 @@ datasets:
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# mGPT-gpt-pierre_DS-msgs-df_Ep-2_Bs-4
+# mGPT: fine-tune on message data MWE
 
-This model is a fine-tuned version of [sberbank-ai/mGPT](https://huggingface.co/sberbank-ai/mGPT) on the None dataset.
+This model is a fine-tuned version of [sberbank-ai/mGPT](https://huggingface.co/sberbank-ai/mGPT) on 80k messages. Trained for one epoch, will be updated in a (separate) model repo later.
 
 ## Model description
 
@@ -48,7 +48,7 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine_with_restarts
 - lr_scheduler_warmup_ratio: 0.05
-- num_epochs: 2
+- num_epochs: 1
 
 ### Framework versions