End of training

README.md

@@ -23,7 +23,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.8369565217391305
+      value: 0.7967479674796748
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -33,8 +33,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the audiofolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6682
-- Accuracy: 0.8370
+- Loss: 0.8032
+- Accuracy: 0.7967
 
 ## Model description
 
@@ -61,14 +61,16 @@ The following hyperparameters were used during training:
 - total_train_batch_size: 64
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: cosine
-- num_epochs: 2
+- num_epochs: 4
 
 ### Training results
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:--------:|
-| No log        | 0.8696 | 5    | 0.6812          | 0.8370   |
-| No log        | 1.7391 | 10   | 0.6682          | 0.8370   |
+| No log        | 0.9032 | 7    | 1.0528          | 0.7317   |
+| No log        | 1.9355 | 15   | 0.9343          | 0.7724   |
+| No log        | 2.9677 | 23   | 0.8097          | 0.7967   |
+| No log        | 3.6129 | 28   | 0.8032          | 0.7967   |
 
 
 ### Framework versions

config.json

@@ -49,20 +49,24 @@
   "hidden_dropout": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "belly_pain",
-    "1": "burping",
-    "2": "discomfort",
-    "3": "hungry",
-    "4": "tired"
+    "0": "asphyxia",
+    "1": "belly_pain",
+    "2": "burping",
+    "3": "discomfort",
+    "4": "hungry",
+    "5": "pain",
+    "6": "tired"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "belly_pain": "0",
-    "burping": "1",
-    "discomfort": "2",
-    "hungry": "3",
-    "tired": "4"
+    "asphyxia": "0",
+    "belly_pain": "1",
+    "burping": "2",
+    "discomfort": "3",
+    "hungry": "4",
+    "pain": "5",
+    "tired": "6"
   },
   "layer_norm_eps": 1e-05,
   "layerdrop": 0.0,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:518cfbb7e0b0273a041bf9f389c2e399583fd8d6dc466f5a9a47f94dac442b10
-size 94766588
+oid sha256:1ddd198ecc5196b8732113a640b9c335237a01045cfc3f0e70a102115fcfe48b
+size 94768644

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b0c541b3ba015fc051ef1bad0e1c8c11050892f7169ee50732a46dddee6589ea
+oid sha256:d2bb14aab55aded97d97bdff2167f490a8f2ab6aafaa41b05ae869002f66a723
 size 5176