Elytum commited on
Commit
191e0ed
1 Parent(s): 1662a63

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.3078
20
- - Accuracy: 0.9031
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: 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
@@ -48,8 +48,8 @@ The following hyperparameters were used during training:
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:-----:|:---------------:|:--------:|
51
- | 0.3282 | 1.0 | 38170 | 0.3190 | 0.8992 |
52
- | 0.3325 | 2.0 | 76340 | 0.3078 | 0.9031 |
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.2083
20
+ - Accuracy: 0.9365
21
 
22
  ## Model description
23
 
 
37
 
38
  The following hyperparameters were used during training:
39
  - learning_rate: 2e-05
40
+ - train_batch_size: 64
41
+ - eval_batch_size: 128
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.2304 | 1.0 | 45257 | 0.2162 | 0.9343 |
52
+ | 0.2181 | 2.0 | 90514 | 0.2083 | 0.9365 |
53
 
54
 
55
  ### Framework versions