update model card README.md

README.md

@@ -16,9 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [prajjwal1/bert-tiny](https://huggingface.co/prajjwal1/bert-tiny) on the lex_glue dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2669
-- Macro-f1: 0.4193
-- Micro-f1: 0.5573
+- Loss: 0.2487
+- Macro-f1: 0.4052
+- Micro-f1: 0.5660
 
 ## Model description
 
@@ -43,17 +43,15 @@ 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
-- num_epochs: 2
+- num_epochs: 1
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Macro-f1 | Micro-f1 |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|
-| 0.2127        | 0.44  | 500  | 0.3104          | 0.3108   | 0.4418   |
-| 0.1808        | 0.89  | 1000 | 0.2961          | 0.3452   | 0.5022   |
-| 0.1762        | 1.33  | 1500 | 0.2807          | 0.3945   | 0.5273   |
-| 0.16          | 1.78  | 2000 | 0.2669          | 0.4193   | 0.5573   |
+| 0.2056        | 0.44  | 500  | 0.2846          | 0.3335   | 0.4763   |
+| 0.1698        | 0.89  | 1000 | 0.2487          | 0.4052   | 0.5660   |
 
 
 ### Framework versions