Hugging Face's logo

Hemg
/
dangerous-heartbeat-detection

Audio Classification
Transformers
TensorBoard
Safetensors
wav2vec2
Generated from Trainer
Eval Results (legacy)
Model card Files
xet
 Metrics Community
Hemg commited on
Commit
5b157a8
·
verified ·
1 Parent(s): 8f69d14

Model save

Browse files
Files changed (2) hide show
  1. README.md +7 -7
  2. model.safetensors +1 -1
README.md CHANGED
@@ -32,7 +32,7 @@ should probably proofread and complete it, then remove this comment. -->
32
 
33
  This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the audiofolder dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.2750
36
  - Accuracy: 1.0
37
 
38
  ## Model description
@@ -52,7 +52,7 @@ More information needed
52
  ### Training hyperparameters
53
 
54
  The following hyperparameters were used during training:
55
- - learning_rate: 0.0003
56
  - train_batch_size: 8
57
  - eval_batch_size: 8
58
  - seed: 42
@@ -60,17 +60,17 @@ The following hyperparameters were used during training:
60
  - total_train_batch_size: 32
61
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
62
  - lr_scheduler_type: linear
63
- - lr_scheduler_warmup_ratio: 0.3
64
  - num_epochs: 4
65
 
66
  ### Training results
67
 
68
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
69
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
70
- | 1.7738 | 0.89 | 2 | 1.3198 | 1.0 |
71
- | 1.1178 | 1.78 | 4 | 0.5868 | 1.0 |
72
- | 0.526 | 2.67 | 6 | 0.3511 | 1.0 |
73
- | 0.2101 | 3.56 | 8 | 0.2750 | 1.0 |
74
 
75
 
76
  ### Framework versions
 
32
 
33
  This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the audiofolder dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.1647
36
  - Accuracy: 1.0
37
 
38
  ## Model description
 
52
  ### Training hyperparameters
53
 
54
  The following hyperparameters were used during training:
55
+ - learning_rate: 0.0005
56
  - train_batch_size: 8
57
  - eval_batch_size: 8
58
  - seed: 42
 
60
  - total_train_batch_size: 32
61
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
62
  - lr_scheduler_type: linear
63
+ - lr_scheduler_warmup_ratio: 0.5
64
  - num_epochs: 4
65
 
66
  ### Training results
67
 
68
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
69
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
70
+ | 1.7573 | 0.89 | 2 | 1.2997 | 1.0 |
71
+ | 1.0518 | 1.78 | 4 | 0.5840 | 1.0 |
72
+ | 0.4723 | 2.67 | 6 | 0.2595 | 1.0 |
73
+ | 0.1425 | 3.56 | 8 | 0.1647 | 1.0 |
74
 
75
 
76
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c00edf14b6a4fff451bb302f1e87de0252cc9bf74605cfdedbd615b3c744bc49
3
  size 378305452
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88f42fa3bcb254a3b6cd089dcfdcebfdb1e1a0bfa95ee5121830738aca8dd5cc
3
  size 378305452