Update README.md

README.md

@@ -11,7 +11,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # ai-club-inductions-21-nlp-ELECTRA-base-squad
 
-This model was trained from scratch on an unknown dataset.
+This model is the deepset/electra-base-squad2 pre-trained model trained on data from AI Inductions 21 NLP competition (https://www.kaggle.com/c/ai-inductions-21-nlp) for extractive QA.
 
 ## Model description
 
@@ -19,18 +19,21 @@ More information needed
 
 ## Intended uses & limitations
 
-More information needed
+AI Inductions 21 NLP competition
 
 ## Training and evaluation data
 
-More information needed
+AI Inductions 21 NLP competition data
 
 ## Training procedure
 
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
+- max_length = 512
+- doc_stride = 384
 - learning_rate: 2e-05
+- weight_decay=0.01
 - train_batch_size: 16
 - eval_batch_size: 16
 - seed: 42