Hugging Face's logo

timothy-geiger
/
distilhubert-finetuned-gtzan

Audio Classification
Transformers
PyTorch
TensorBoard
Safetensors
hubert
Generated from Trainer
Eval Results
Inference Endpoints
Model card Files Metrics Community
timothy-geiger commited on
Commit
ad2625e
•
1 Parent(s): c8bd9d9

End of training

Browse files
Files changed (2) hide show
  1. README.md +17 -17
  2. model.safetensors +1 -1
README.md CHANGED
@@ -32,7 +32,7 @@ should probably proofread and complete it, then remove this comment. -->
32
 
33
  This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the GTZAN dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.9578
36
  - Accuracy: 0.81
37
 
38
  ## Model description
@@ -52,7 +52,7 @@ More information needed
52
  ### Training hyperparameters
53
 
54
  The following hyperparameters were used during training:
55
- - learning_rate: 5e-05
56
  - train_batch_size: 8
57
  - eval_batch_size: 8
58
  - seed: 42
@@ -66,21 +66,21 @@ The following hyperparameters were used during training:
66
 
67
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
68
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
69
- | 2.0873 | 1.0 | 113 | 2.0168 | 0.5 |
70
- | 1.3586 | 2.0 | 226 | 1.3771 | 0.6 |
71
- | 1.0737 | 3.0 | 339 | 1.0985 | 0.71 |
72
- | 0.7844 | 4.0 | 452 | 0.9615 | 0.7 |
73
- | 0.7443 | 5.0 | 565 | 0.8968 | 0.72 |
74
- | 0.3561 | 6.0 | 678 | 0.7699 | 0.78 |
75
- | 0.2659 | 7.0 | 791 | 0.7783 | 0.78 |
76
- | 0.0942 | 8.0 | 904 | 0.7705 | 0.8 |
77
- | 0.1212 | 9.0 | 1017 | 0.8313 | 0.83 |
78
- | 0.0189 | 10.0 | 1130 | 0.8504 | 0.82 |
79
- | 0.0126 | 11.0 | 1243 | 0.9311 | 0.82 |
80
- | 0.0101 | 12.0 | 1356 | 0.9227 | 0.81 |
81
- | 0.0081 | 13.0 | 1469 | 0.9378 | 0.81 |
82
- | 0.0079 | 14.0 | 1582 | 0.9390 | 0.82 |
83
- | 0.0065 | 15.0 | 1695 | 0.9578 | 0.81 |
84
 
85
 
86
  ### Framework versions
 
32
 
33
  This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the GTZAN dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.6734
36
  - Accuracy: 0.81
37
 
38
  ## Model description
 
52
  ### Training hyperparameters
53
 
54
  The following hyperparameters were used during training:
55
+ - learning_rate: 3e-05
56
  - train_batch_size: 8
57
  - eval_batch_size: 8
58
  - seed: 42
 
66
 
67
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
68
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
69
+ | 2.1966 | 1.0 | 113 | 2.1508 | 0.49 |
70
+ | 1.6173 | 2.0 | 226 | 1.6236 | 0.57 |
71
+ | 1.4483 | 3.0 | 339 | 1.3165 | 0.69 |
72
+ | 1.1295 | 4.0 | 452 | 1.1380 | 0.7 |
73
+ | 0.9444 | 5.0 | 565 | 0.9592 | 0.76 |
74
+ | 0.8205 | 6.0 | 678 | 0.9024 | 0.74 |
75
+ | 0.6711 | 7.0 | 791 | 0.8497 | 0.78 |
76
+ | 0.4759 | 8.0 | 904 | 0.8240 | 0.78 |
77
+ | 0.4657 | 9.0 | 1017 | 0.7534 | 0.79 |
78
+ | 0.3651 | 10.0 | 1130 | 0.7403 | 0.8 |
79
+ | 0.3154 | 11.0 | 1243 | 0.7082 | 0.82 |
80
+ | 0.252 | 12.0 | 1356 | 0.7222 | 0.81 |
81
+ | 0.2265 | 13.0 | 1469 | 0.7063 | 0.82 |
82
+ | 0.2478 | 14.0 | 1582 | 0.6898 | 0.81 |
83
+ | 0.1386 | 15.0 | 1695 | 0.6734 | 0.81 |
84
 
85
 
86
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6c5d18c2033ebe6ef735f64b028e437abbf1965f416ca439b8082bbc287a5545
3
  size 94771728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99e1756baaf7bd573b62920e84275c4451f0dd17809d50f606d9697f5dde44e0
3
  size 94771728