Elytum commited on
Commit
5988235
1 Parent(s): bbd6c2b

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -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 [prajjwal1/bert-tiny](https://huggingface.co/prajjwal1/bert-tiny) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.4543
20
- - Accuracy: 0.8780
21
 
22
  ## Model description
23
 
@@ -37,8 +37,8 @@ More information needed
37
 
38
  The following hyperparameters were used during training:
39
  - learning_rate: 2e-05
40
- - train_batch_size: 8
41
- - eval_batch_size: 8
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
@@ -48,8 +48,8 @@ The following hyperparameters were used during training:
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
- | 0.7941 | 1.0 | 4929 | 0.5998 | 0.8557 |
52
- | 0.5703 | 2.0 | 9858 | 0.4543 | 0.8780 |
53
 
54
 
55
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [prajjwal1/bert-tiny](https://huggingface.co/prajjwal1/bert-tiny) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.3382
20
+ - Accuracy: 0.9019
21
 
22
  ## Model description
23
 
 
37
 
38
  The following hyperparameters were used during training:
39
  - learning_rate: 2e-05
40
+ - train_batch_size: 128
41
+ - eval_batch_size: 256
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
 
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
+ | 0.5322 | 1.0 | 2865 | 0.4202 | 0.8942 |
52
+ | 0.3652 | 2.0 | 5730 | 0.3382 | 0.9019 |
53
 
54
 
55
  ### Framework versions