SHIVA SINGH BAGRI commited on
Commit
d324824
·
1 Parent(s): 3433fc0

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -10
README.md CHANGED
@@ -1,7 +1,5 @@
1
  ---
2
- library_name: transformers
3
  license: apache-2.0
4
- base_model: bert-base-uncased
5
  tags:
6
  - generated_from_trainer
7
  metrics:
@@ -18,8 +16,8 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.0814
22
- - Accuracy: 0.9826
23
 
24
  ## Model description
25
 
@@ -42,7 +40,7 @@ The following hyperparameters were used during training:
42
  - train_batch_size: 16
43
  - eval_batch_size: 16
44
  - seed: 42
45
- - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
46
  - lr_scheduler_type: linear
47
  - num_epochs: 2
48
 
@@ -50,13 +48,13 @@ The following hyperparameters were used during training:
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
- | 0.0409 | 1.0 | 689 | 0.1486 | 0.9666 |
54
- | 0.0055 | 2.0 | 1378 | 0.0814 | 0.9826 |
55
 
56
 
57
  ### Framework versions
58
 
59
- - Transformers 4.50.0
60
- - Pytorch 2.6.0+cu124
61
  - Datasets 3.4.1
62
- - Tokenizers 0.21.1
 
1
  ---
 
2
  license: apache-2.0
 
3
  tags:
4
  - generated_from_trainer
5
  metrics:
 
16
 
17
  This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.1438
20
+ - Accuracy: 0.9649
21
 
22
  ## Model description
23
 
 
40
  - train_batch_size: 16
41
  - eval_batch_size: 16
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
 
 
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
+ | 0.069 | 1.0 | 506 | 0.1066 | 0.9693 |
52
+ | 0.0174 | 2.0 | 1012 | 0.1438 | 0.9649 |
53
 
54
 
55
  ### Framework versions
56
 
57
+ - Transformers 4.28.0
58
+ - Pytorch 2.2.1+cu121
59
  - Datasets 3.4.1
60
+ - Tokenizers 0.13.3