zayedupal commited on
Commit
701c86a
·
1 Parent(s): 4f91101

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -6
README.md CHANGED
@@ -16,8 +16,8 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 1.6774
20
- - Accuracy: 0.4038
21
 
22
  ## Model description
23
 
@@ -42,19 +42,22 @@ The following hyperparameters were used during training:
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
- - num_epochs: 2
46
 
47
  ### Training results
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
- | No log | 1.0 | 189 | 1.7138 | 0.3960 |
52
- | No log | 2.0 | 378 | 1.6774 | 0.4038 |
 
 
 
53
 
54
 
55
  ### Framework versions
56
 
57
  - Transformers 4.24.0
58
  - Pytorch 2.0.1+cu118
59
- - Datasets 2.12.0
60
  - Tokenizers 0.13.3
 
16
 
17
  This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 1.6396
20
+ - Accuracy: 0.4191
21
 
22
  ## Model description
23
 
 
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
+ - num_epochs: 5
46
 
47
  ### Training results
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
+ | No log | 1.0 | 189 | 1.6919 | 0.4019 |
52
+ | No log | 2.0 | 378 | 1.6525 | 0.4173 |
53
+ | 1.718 | 3.0 | 567 | 1.6504 | 0.4153 |
54
+ | 1.718 | 4.0 | 756 | 1.6403 | 0.4196 |
55
+ | 1.718 | 5.0 | 945 | 1.6396 | 0.4191 |
56
 
57
 
58
  ### Framework versions
59
 
60
  - Transformers 4.24.0
61
  - Pytorch 2.0.1+cu118
62
+ - Datasets 2.13.1
63
  - Tokenizers 0.13.3