Elytum commited on
Commit
f491fcb
1 Parent(s): 4c8a7fc

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- license: mit
3
  tags:
4
  - generated_from_trainer
5
  metrics:
@@ -14,10 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  # tiny-classification-fast-2
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: 1.9520
20
- - Accuracy: 0.6111
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: 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,8 +48,8 @@ The following hyperparameters were used during training:
48
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
- | No log | 1.0 | 271 | 2.0859 | 0.5593 |
52
- | 2.1669 | 2.0 | 542 | 1.9520 | 0.6111 |
53
 
54
 
55
  ### Framework versions
 
1
  ---
2
+ license: apache-2.0
3
  tags:
4
  - generated_from_trainer
5
  metrics:
 
14
 
15
  # tiny-classification-fast-2
16
 
17
+ This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 0.7317
20
+ - Accuracy: 0.8089
21
 
22
  ## Model description
23
 
 
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
 
49
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
50
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
51
+ | 0.3759 | 1.0 | 1021 | 0.6279 | 0.7853 |
52
+ | 0.2043 | 2.0 | 2042 | 0.7317 | 0.8089 |
53
 
54
 
55
  ### Framework versions