chin-may commited on
Commit
af09364
1 Parent(s): d941059

End of training

Browse files
Files changed (1) hide show
  1. README.md +17 -8
README.md CHANGED
@@ -3,6 +3,8 @@ license: apache-2.0
3
  base_model: facebook/wav2vec2-base
4
  tags:
5
  - generated_from_trainer
 
 
6
  model-index:
7
  - name: wav2vec2-audio-emotion-classification
8
  results: []
@@ -15,13 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - eval_loss: 1.7242
19
- - eval_accuracy: 0.3958
20
- - eval_runtime: 20.3868
21
- - eval_samples_per_second: 65.189
22
- - eval_steps_per_second: 2.06
23
- - epoch: 22.99
24
- - step: 960
25
 
26
  ## Model description
27
 
@@ -49,7 +46,19 @@ The following hyperparameters were used during training:
49
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
50
  - lr_scheduler_type: linear
51
  - lr_scheduler_warmup_ratio: 0.1
52
- - num_epochs: 30
 
 
 
 
 
 
 
 
 
 
 
 
53
 
54
  ### Framework versions
55
 
 
3
  base_model: facebook/wav2vec2-base
4
  tags:
5
  - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
  model-index:
9
  - name: wav2vec2-audio-emotion-classification
10
  results: []
 
17
 
18
  This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.9518
21
+ - Accuracy: 0.7398
 
 
 
 
 
22
 
23
  ## Model description
24
 
 
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_ratio: 0.1
49
+ - num_epochs: 6
50
+
51
+ ### Training results
52
+
53
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
+ | 1.759 | 0.99 | 22 | 1.7087 | 0.3122 |
56
+ | 1.5568 | 1.98 | 44 | 1.4412 | 0.4923 |
57
+ | 1.2577 | 2.97 | 66 | 1.1467 | 0.7060 |
58
+ | 1.0768 | 4.0 | 89 | 1.0131 | 0.7215 |
59
+ | 0.9476 | 4.99 | 111 | 0.9633 | 0.7314 |
60
+ | 0.9094 | 5.93 | 132 | 0.9518 | 0.7398 |
61
+
62
 
63
  ### Framework versions
64