End of training

README.md

@@ -48,7 +48,7 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:--------:|
-| No log        | 0.9888 | 11   | 1.6447          | 0.6067   |
+| No log        | 0.9888 | 11   | 0.3579          | 0.8764   |
 
 
 ### Framework versions

config.json

@@ -49,28 +49,14 @@
   "hidden_dropout": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "Baby laugh",
-    "1": "Noise",
-    "2": "Silence",
-    "3": "belly_pain",
-    "4": "burping",
-    "5": "crying_baby",
-    "6": "discomfort",
-    "7": "hungry",
-    "8": "tired"
+    "0": "no_crying",
+    "1": "crying"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "Baby laugh": "0",
-    "Noise": "1",
-    "Silence": "2",
-    "belly_pain": "3",
-    "burping": "4",
-    "crying_baby": "5",
-    "discomfort": "6",
-    "hungry": "7",
-    "tired": "8"
+    "crying": "1",
+    "no_crying": "0"
   },
   "layer_norm_eps": 1e-05,
   "layerdrop": 0.0,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:de07a99b3258fafebbf992417c78da910bc64a4fce8471341c51d56ccaaea51d
-size 94770700
+oid sha256:bc63aa946031b115f9f87d7c3684a6364bb68eed3cea79c726bfb3a94e1bd28f
+size 94763496

runs/Aug13_13-31-13_DesMar/events.out.tfevents.1723548681.DesMar.27060.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6a459f9e9bec4d2b0a19fd51db2e9039c2c1ea609a942758cb382f1d6289dd0d
+size 6521

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ecce46da85347cfd90be4aab977a45410450b537293bd9dbee293dd9bbf1725f
+oid sha256:be7f88858fe87d88b44aa54ed255bfc37d0f2646bc90a45a75cda85089e3b791
 size 5240