sfedar commited on
Commit
99a36b5
1 Parent(s): 2bafbd1

End of training, 9 epochs, 4 batch size, writer batch size: 1000, 1 gradient accumulation steps, learning rate: 5e-05, 30 s

Browse files
Files changed (2) hide show
  1. README.md +10 -10
  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.6050
36
  - Accuracy: 0.82
37
 
38
  ## Model description
@@ -66,15 +66,15 @@ The following hyperparameters were used during training:
66
 
67
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
68
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
69
- | 1.9557 | 1.0 | 113 | 1.9166 | 0.41 |
70
- | 1.1928 | 2.0 | 226 | 1.2556 | 0.67 |
71
- | 1.0954 | 3.0 | 339 | 1.0473 | 0.72 |
72
- | 0.6237 | 4.0 | 452 | 0.8854 | 0.72 |
73
- | 0.4856 | 5.0 | 565 | 0.6624 | 0.84 |
74
- | 0.344 | 6.0 | 678 | 0.6091 | 0.82 |
75
- | 0.2819 | 7.0 | 791 | 0.6306 | 0.8 |
76
- | 0.1515 | 8.0 | 904 | 0.5855 | 0.82 |
77
- | 0.2444 | 9.0 | 1017 | 0.6050 | 0.82 |
78
 
79
 
80
  ### 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.5819
36
  - Accuracy: 0.82
37
 
38
  ## Model description
 
66
 
67
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
68
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
69
+ | 1.9236 | 1.0 | 113 | 1.8095 | 0.51 |
70
+ | 1.1771 | 2.0 | 226 | 1.2024 | 0.66 |
71
+ | 1.0227 | 3.0 | 339 | 1.0044 | 0.71 |
72
+ | 0.6925 | 4.0 | 452 | 0.8156 | 0.76 |
73
+ | 0.5823 | 5.0 | 565 | 0.6761 | 0.79 |
74
+ | 0.3919 | 6.0 | 678 | 0.6031 | 0.79 |
75
+ | 0.3968 | 7.0 | 791 | 0.6003 | 0.83 |
76
+ | 0.1533 | 8.0 | 904 | 0.5785 | 0.81 |
77
+ | 0.2358 | 9.0 | 1017 | 0.5819 | 0.82 |
78
 
79
 
80
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:efb42301ff4892fd70b1a89893b5ab051aba8de6936d844f20ec38df37fbaafa
3
  size 94771728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4017611a80088dba6bfd80bf2b6f1ba9bc85d07a6a19b611c2e2c48ac5041235
3
  size 94771728