shtif commited on
Commit
3f35182
1 Parent(s): 7b8d24f

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -7
README.md CHANGED
@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the GTZAN dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.7706
22
- - Accuracy: 0.84
23
 
24
  ## Model description
25
 
@@ -38,7 +38,7 @@ More information needed
38
  ### Training hyperparameters
39
 
40
  The following hyperparameters were used during training:
41
- - learning_rate: 8e-05
42
  - train_batch_size: 8
43
  - eval_batch_size: 8
44
  - seed: 42
@@ -52,10 +52,10 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
- | 0.0486 | 1.0 | 113 | 0.9972 | 0.8 |
56
- | 0.0018 | 2.0 | 226 | 0.8953 | 0.81 |
57
- | 0.0025 | 3.0 | 339 | 0.7969 | 0.83 |
58
- | 0.0007 | 4.0 | 452 | 0.7706 | 0.84 |
59
 
60
 
61
  ### Framework versions
 
18
 
19
  This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the GTZAN dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.8798
22
+ - Accuracy: 0.83
23
 
24
  ## Model description
25
 
 
38
  ### Training hyperparameters
39
 
40
  The following hyperparameters were used during training:
41
+ - learning_rate: 9e-05
42
  - train_batch_size: 8
43
  - eval_batch_size: 8
44
  - seed: 42
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
+ | 0.4135 | 1.0 | 113 | 1.1206 | 0.84 |
56
+ | 0.0018 | 2.0 | 226 | 1.4291 | 0.8 |
57
+ | 0.0004 | 3.0 | 339 | 1.1077 | 0.83 |
58
+ | 0.0002 | 4.0 | 452 | 0.8798 | 0.83 |
59
 
60
 
61
  ### Framework versions