Update README.md

README.md

@@ -20,8 +20,6 @@ It achieves the following results on the evaluation set:
 ## Model description
 - Base Model: bert-base-uncased
 - Block_size: 512
-- Batch_size: 8
-- Num_epochs: 10
 
 ## Intended uses & limitations
 
@@ -29,7 +27,7 @@ More information needed
 
 ## Training and evaluation data
 
-- CCHS visit-level texts
+- MAYO visit-level texts
 
 ## Training procedure
 
@@ -38,6 +36,9 @@ More information needed
 The following hyperparameters were used during training:
 - optimizer: {'name': 'AdamWeightDecay', 'learning_rate': 2e-05, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False, 'weight_decay_rate': 0.01}
 - training_precision: float32
+- mlm_probability: 0.15
+- batch_size: 8
+- epochs: 10
 
 ### Training results