ykirpichev commited on
Commit
eb172b9
1 Parent(s): ad52c3e

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -20,7 +20,7 @@ model-index:
20
  metrics:
21
  - name: Accuracy
22
  type: accuracy
23
- value: 0.85
24
  ---
25
 
26
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,8 +30,8 @@ should probably proofread and complete it, then remove this comment. -->
30
 
31
  This model is a fine-tuned version of [ykirpichev/distilhubert-finetuned-gtzan](https://huggingface.co/ykirpichev/distilhubert-finetuned-gtzan) on the GTZAN dataset.
32
  It achieves the following results on the evaluation set:
33
- - Loss: 0.6923
34
- - Accuracy: 0.85
35
 
36
  ## Model description
37
 
@@ -63,11 +63,11 @@ The following hyperparameters were used during training:
63
 
64
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
65
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
66
- | 0.0418 | 1.0 | 113 | 0.5770 | 0.84 |
67
- | 0.2014 | 2.0 | 226 | 0.7910 | 0.82 |
68
- | 0.1594 | 3.0 | 339 | 0.7139 | 0.85 |
69
- | 0.0037 | 4.0 | 452 | 0.6513 | 0.89 |
70
- | 0.0031 | 5.0 | 565 | 0.6923 | 0.85 |
71
 
72
 
73
  ### Framework versions
 
20
  metrics:
21
  - name: Accuracy
22
  type: accuracy
23
+ value: 0.83
24
  ---
25
 
26
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
30
 
31
  This model is a fine-tuned version of [ykirpichev/distilhubert-finetuned-gtzan](https://huggingface.co/ykirpichev/distilhubert-finetuned-gtzan) on the GTZAN dataset.
32
  It achieves the following results on the evaluation set:
33
+ - Loss: 1.0892
34
+ - Accuracy: 0.83
35
 
36
  ## Model description
37
 
 
63
 
64
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
65
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
66
+ | 0.0171 | 1.0 | 113 | 1.8133 | 0.73 |
67
+ | 0.0637 | 2.0 | 226 | 1.3377 | 0.79 |
68
+ | 0.2052 | 3.0 | 339 | 0.8646 | 0.88 |
69
+ | 0.0019 | 4.0 | 452 | 1.0868 | 0.82 |
70
+ | 0.0003 | 5.0 | 565 | 1.0892 | 0.83 |
71
 
72
 
73
  ### Framework versions