Hemg commited on
Commit
5ca1531
1 Parent(s): 0930b62

Model save

Browse files
Files changed (2) hide show
  1. README.md +6 -6
  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 [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the audiofolder dataset.
34
  It achieves the following results on the evaluation set:
35
- - Loss: 0.0023
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
@@ -65,10 +65,10 @@ The following hyperparameters were used during training:
65
 
66
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
- | 1.0263 | 1.0 | 8 | 0.2218 | 1.0 |
69
- | 0.104 | 2.0 | 16 | 0.0153 | 1.0 |
70
- | 0.0099 | 3.0 | 24 | 0.0034 | 1.0 |
71
- | 0.0036 | 4.0 | 32 | 0.0023 | 1.0 |
72
 
73
 
74
  ### Framework versions
 
32
 
33
  This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the audiofolder dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.0140
36
  - Accuracy: 1.0
37
 
38
  ## Model description
 
52
  ### Training hyperparameters
53
 
54
  The following hyperparameters were used during training:
55
+ - learning_rate: 0.0002
56
  - train_batch_size: 8
57
  - eval_batch_size: 8
58
  - seed: 42
 
65
 
66
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
+ | 1.1307 | 1.0 | 8 | 0.3556 | 1.0 |
69
+ | 0.2175 | 2.0 | 16 | 0.0685 | 1.0 |
70
+ | 0.0491 | 3.0 | 24 | 0.0205 | 1.0 |
71
+ | 0.0208 | 4.0 | 32 | 0.0140 | 1.0 |
72
 
73
 
74
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:64cd0eb094be8363da0b6e28b21b6c88291f4ee1ed0b292d4bf4c51569e3c257
3
  size 94766588
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c9e951812c5002f5bd215711f39ab1b0ba47a03fa658dd879c7de5d4a4f60f17
3
  size 94766588