Hemg commited on
Commit
3aaaf3c
1 Parent(s): c4d5d5d

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.0140
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.0002
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.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
 
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: 1.2548
36
  - Accuracy: 1.0
37
 
38
  ## Model description
 
52
  ### Training hyperparameters
53
 
54
  The following hyperparameters were used during training:
55
+ - learning_rate: 1e-05
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.591 | 1.0 | 8 | 1.5051 | 0.8519 |
69
+ | 1.4364 | 2.0 | 16 | 1.3666 | 1.0 |
70
+ | 1.3203 | 3.0 | 24 | 1.2839 | 1.0 |
71
+ | 1.263 | 4.0 | 32 | 1.2548 | 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:597ef4c7dda7a4ca58eacf05b0a903bd2703b168ec4beecab743be2a2714302b
3
  size 94766588
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90c83170e9cc7f5f7126582023a1872fe0d9013b3aa5b255f9af0c5c5694a8fa
3
  size 94766588