ditwoo commited on
Commit
e419880
1 Parent(s): d304330

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -14
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 [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the GTZAN dataset.
20
- It achieves the following results on the evaluation set:
21
- - Loss: 1.1300
22
- - Accuracy: 0.68
23
 
24
  ## Model description
25
 
@@ -45,15 +40,7 @@ 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: 2
49
-
50
- ### Training results
51
-
52
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
- |:-------------:|:-----:|:----:|:---------------:|:--------:|
54
- | 1.752 | 1.0 | 225 | 1.4066 | 0.58 |
55
- | 0.9226 | 2.0 | 450 | 1.1300 | 0.68 |
56
-
57
 
58
  ### Framework versions
59
 
 
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 [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) 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: 30
 
 
 
 
 
 
 
 
44
 
45
  ### Framework versions
46