mcamara commited on
Commit
92cf1e4
1 Parent(s): 8421444

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -6
README.md CHANGED
@@ -4,6 +4,8 @@ tags:
4
  - generated_from_trainer
5
  datasets:
6
  - marsyas/gtzan
 
 
7
  model-index:
8
  - name: distilhubert-finetuned-gtzan
9
  results: []
@@ -16,12 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [MariaK/distilhubert-finetuned-gtzan-v2](https://huggingface.co/MariaK/distilhubert-finetuned-gtzan-v2) on the GTZAN dataset.
18
  It achieves the following results on the evaluation set:
19
- - eval_loss: 0.4883
20
- - eval_accuracy: 0.87
21
- - eval_runtime: 25.6473
22
- - eval_samples_per_second: 3.899
23
- - eval_steps_per_second: 0.507
24
- - step: 0
25
 
26
  ## Model description
27
 
@@ -49,6 +47,13 @@ The following hyperparameters were used during training:
49
  - lr_scheduler_warmup_ratio: 0.1
50
  - num_epochs: 0.01
51
 
 
 
 
 
 
 
 
52
  ### Framework versions
53
 
54
  - Transformers 4.30.2
 
4
  - generated_from_trainer
5
  datasets:
6
  - marsyas/gtzan
7
+ metrics:
8
+ - accuracy
9
  model-index:
10
  - name: distilhubert-finetuned-gtzan
11
  results: []
 
18
 
19
  This model is a fine-tuned version of [MariaK/distilhubert-finetuned-gtzan-v2](https://huggingface.co/MariaK/distilhubert-finetuned-gtzan-v2) on the GTZAN dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.4905
22
+ - Accuracy: 0.89
 
 
 
 
23
 
24
  ## Model description
25
 
 
47
  - lr_scheduler_warmup_ratio: 0.1
48
  - num_epochs: 0.01
49
 
50
+ ### Training results
51
+
52
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
54
+ | 0.0742 | 0.02 | 2 | 0.4905 | 0.89 |
55
+
56
+
57
  ### Framework versions
58
 
59
  - Transformers 4.30.2