update model card README.md

README.md

@@ -1,6 +1,5 @@
 ---
-language:
-- en
+license: apache-2.0
 tags:
 - generated_from_trainer
 datasets:
@@ -14,13 +13,15 @@ model-index:
       name: Text Classification
       type: text-classification
     dataset:
-      name: GLUE COLA
+      name: glue
       type: glue
+      config: cola
+      split: validation
       args: cola
     metrics:
     - name: Matthews Correlation
       type: matthews_correlation
-      value: 0.0
+      value: 0.08118499547243287
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -28,10 +29,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # mobilebert_sa_GLUE_Experiment_cola
 
-This model is a fine-tuned version of [](https://huggingface.co/) on the GLUE COLA dataset.
+This model is a fine-tuned version of [google/mobilebert-uncased](https://huggingface.co/google/mobilebert-uncased) on the glue dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6131
-- Matthews Correlation: 0.0
+- Loss: 0.6915
+- Matthews Correlation: 0.0812
 
 ## Model description
 
@@ -51,8 +52,8 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
-- train_batch_size: 256
-- eval_batch_size: 256
+- train_batch_size: 128
+- eval_batch_size: 128
 - seed: 10
 - distributed_type: multi-GPU
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
@@ -64,23 +65,20 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Matthews Correlation |
 |:-------------:|:-----:|:----:|:---------------:|:--------------------:|
-| 0.6197        | 1.0   | 34   | 0.6239          | 0.0                  |
-| 0.6078        | 2.0   | 68   | 0.6179          | 0.0                  |
-| 0.6064        | 3.0   | 102  | 0.6180          | 0.0                  |
-| 0.6073        | 4.0   | 136  | 0.6176          | 0.0                  |
-| 0.6069        | 5.0   | 170  | 0.6173          | 0.0                  |
-| 0.6043        | 6.0   | 204  | 0.6166          | 0.0                  |
-| 0.6004        | 7.0   | 238  | 0.6131          | 0.0                  |
-| 0.5842        | 8.0   | 272  | 0.6241          | 0.0951               |
-| 0.5192        | 9.0   | 306  | 0.6362          | 0.0598               |
-| 0.4884        | 10.0  | 340  | 0.7010          | 0.0801               |
-| 0.4559        | 11.0  | 374  | 0.6731          | 0.0905               |
-| 0.4367        | 12.0  | 408  | 0.6893          | 0.0901               |
+| 0.6122        | 1.0   | 67   | 0.6184          | 0.0                  |
+| 0.6078        | 2.0   | 134  | 0.6180          | 0.0                  |
+| 0.607         | 3.0   | 201  | 0.6185          | 0.0                  |
+| 0.6052        | 4.0   | 268  | 0.6153          | 0.0                  |
+| 0.5822        | 5.0   | 335  | 0.6292          | 0.0506               |
+| 0.5193        | 6.0   | 402  | 0.6422          | 0.0743               |
+| 0.4783        | 7.0   | 469  | 0.7020          | 0.0629               |
+| 0.4504        | 8.0   | 536  | 0.7422          | 0.0834               |
+| 0.4315        | 9.0   | 603  | 0.6915          | 0.0812               |
 
 
 ### Framework versions
 
-- Transformers 4.25.1
+- Transformers 4.26.0
 - Pytorch 1.14.0a0+410ce96
 - Datasets 2.8.0
 - Tokenizers 0.13.2