mcamara commited on
Commit
7215f64
1 Parent(s): 9a8e385

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -7
README.md CHANGED
@@ -4,8 +4,6 @@ tags:
4
  - generated_from_trainer
5
  datasets:
6
  - marsyas/gtzan
7
- metrics:
8
- - accuracy
9
  model-index:
10
  - name: distilhubert-finetuned-gtzan
11
  results: []
@@ -17,9 +15,6 @@ should probably proofread and complete it, then remove this comment. -->
17
  # distilhubert-finetuned-gtzan
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.5224
22
- - Accuracy: 0.84
23
 
24
  ## Model description
25
 
@@ -45,13 +40,13 @@ The following hyperparameters were used during training:
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
  - lr_scheduler_warmup_ratio: 0.1
48
- - num_epochs: 0.1
49
 
50
  ### Training results
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
54
- | 0.124 | 0.11 | 12 | 0.5224 | 0.84 |
55
 
56
 
57
  ### Framework versions
 
4
  - generated_from_trainer
5
  datasets:
6
  - marsyas/gtzan
 
 
7
  model-index:
8
  - name: distilhubert-finetuned-gtzan
9
  results: []
 
15
  # distilhubert-finetuned-gtzan
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
 
19
  ## Model description
20
 
 
40
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
41
  - lr_scheduler_type: linear
42
  - lr_scheduler_warmup_ratio: 0.1
43
+ - num_epochs: 0.01
44
 
45
  ### Training results
46
 
47
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
48
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
49
+ | No log | 0.02 | 2 | 0.4905 | 0.89 |
50
 
51
 
52
  ### Framework versions