winegarj commited on
Commit
1e1330c
·
1 Parent(s): bc84d08

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -11
README.md CHANGED
@@ -16,7 +16,7 @@ model-index:
16
  name: glue
17
  type: glue
18
  config: sst2
19
- split: train
20
  args: sst2
21
  metrics:
22
  - name: Accuracy
@@ -31,7 +31,7 @@ should probably proofread and complete it, then remove this comment. -->
31
 
32
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the glue dataset.
33
  It achieves the following results on the evaluation set:
34
- - Loss: 0.2817
35
  - Accuracy: 0.9025
36
 
37
  ## Model description
@@ -63,16 +63,16 @@ The following hyperparameters were used during training:
63
 
64
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
65
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
66
- | No log | 1.0 | 132 | 0.2534 | 0.8956 |
67
- | No log | 2.0 | 264 | 0.2679 | 0.9002 |
68
- | No log | 3.0 | 396 | 0.2817 | 0.9025 |
69
- | 0.1896 | 4.0 | 528 | 0.2933 | 0.9002 |
70
- | 0.1896 | 5.0 | 660 | 0.3008 | 0.9002 |
71
 
72
 
73
  ### Framework versions
74
 
75
- - Transformers 4.23.1
76
- - Pytorch 1.12.1+cu116
77
- - Datasets 2.6.1
78
- - Tokenizers 0.13.1
 
16
  name: glue
17
  type: glue
18
  config: sst2
19
+ split: validation
20
  args: sst2
21
  metrics:
22
  - name: Accuracy
 
31
 
32
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the glue dataset.
33
  It achieves the following results on the evaluation set:
34
+ - Loss: 0.2805
35
  - Accuracy: 0.9025
36
 
37
  ## Model description
 
63
 
64
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
65
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
66
+ | No log | 1.0 | 132 | 0.2592 | 0.8888 |
67
+ | No log | 2.0 | 264 | 0.2770 | 0.8979 |
68
+ | No log | 3.0 | 396 | 0.2805 | 0.9025 |
69
+ | 0.1914 | 4.0 | 528 | 0.2975 | 0.9014 |
70
+ | 0.1914 | 5.0 | 660 | 0.3026 | 0.9002 |
71
 
72
 
73
  ### Framework versions
74
 
75
+ - Transformers 4.29.1
76
+ - Pytorch 2.0.1+cu117
77
+ - Datasets 2.12.0
78
+ - Tokenizers 0.13.3