google_mobilebert-uncased fold 2

README.md

@@ -21,11 +21,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [google/mobilebert-uncased](https://huggingface.co/google/mobilebert-uncased) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: nan
-- Accuracy: 0.5413
-- F1: 0.0
-- Precision: 0.0
-- Recall: 0.0
+- Loss: 0.1152
+- Accuracy: 0.9587
+- F1: 0.9547
+- Precision: 0.9608
+- Recall: 0.9488
 
 ## Model description
 
@@ -51,15 +51,14 @@ The following hyperparameters were used during training:
 - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
 - num_epochs: 3
-- mixed_precision_training: Native AMP
 
 ### Training results
 
-| Training Loss | Epoch | Step  | Validation Loss | Accuracy | F1  | Precision | Recall |
-|:-------------:|:-----:|:-----:|:---------------:|:--------:|:---:|:---------:|:------:|
-| 0.0           | 1.0   | 15481 | nan             | 0.5413   | 0.0 | 0.0       | 0.0    |
-| 0.0           | 2.0   | 30962 | nan             | 0.5413   | 0.0 | 0.0       | 0.0    |
-| 0.0           | 3.0   | 46443 | nan             | 0.5413   | 0.0 | 0.0       | 0.0    |
+| Training Loss | Epoch | Step  | Validation Loss | Accuracy | F1     | Precision | Recall |
+|:-------------:|:-----:|:-----:|:---------------:|:--------:|:------:|:---------:|:------:|
+| 0.1494        | 1.0   | 15481 | 0.1285          | 0.9496   | 0.9445 | 0.9540    | 0.9352 |
+| 0.0827        | 2.0   | 30962 | 0.1849          | 0.9571   | 0.9529 | 0.9586    | 0.9473 |
+| 0.0864        | 3.0   | 46443 | 0.1152          | 0.9587   | 0.9547 | 0.9608    | 0.9488 |
 
 
 ### Framework versions