update model card README.md

README.md

@@ -18,11 +18,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model was trained from scratch on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1370
-- Precision: 0.8146
-- Recall: 0.8349
-- F1: 0.8246
-- Accuracy: 0.9609
+- Loss: 0.1443
+- Precision: 0.8176
+- Recall: 0.8401
+- F1: 0.8287
+- Accuracy: 0.9615
 
 ## Model description
 
@@ -47,17 +47,19 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- lr_scheduler_warmup_steps: 2272
-- num_epochs: 3
+- lr_scheduler_warmup_steps: 3787
+- num_epochs: 5
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch | Step  | Validation Loss | Precision | Recall | F1     | Accuracy |
 |:-------------:|:-----:|:-----:|:---------------:|:---------:|:------:|:------:|:--------:|
-| No log        | 1.0   | 3788  | 0.1367          | 0.8089    | 0.8240 | 0.8164 | 0.9595   |
-| No log        | 2.0   | 7576  | 0.1345          | 0.8130    | 0.8331 | 0.8229 | 0.9604   |
-| 0.0953        | 3.0   | 11364 | 0.1370          | 0.8146    | 0.8349 | 0.8246 | 0.9609   |
+| No log        | 1.0   | 3788  | 0.1511          | 0.8095    | 0.8257 | 0.8176 | 0.9594   |
+| No log        | 2.0   | 7576  | 0.1461          | 0.8121    | 0.8339 | 0.8228 | 0.9600   |
+| No log        | 3.0   | 11364 | 0.1417          | 0.8139    | 0.8372 | 0.8254 | 0.9607   |
+| No log        | 4.0   | 15152 | 0.1418          | 0.8238    | 0.8346 | 0.8292 | 0.9617   |
+| 0.0748        | 5.0   | 18940 | 0.1443          | 0.8176    | 0.8401 | 0.8287 | 0.9615   |
 
 
 ### Framework versions