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.6532          | 0.5506   |
+| No log        | 0.8696 | 5    | 1.0622          | 0.8370   |
 
 
 ### Framework versions

config.json

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

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b20a9c6637a79559507b479b402d7acbb1f95f20c0502d6e6ff4fa53fc3ca03f
-size 94770700
+oid sha256:eed07d78089bf3325d99a4f086ec3dd6acf7a86ac786eb6ece388e96cdaecc6d
+size 94766588

runs/Jul17_20-25-33_DesMar/events.out.tfevents.1721240735.DesMar.8828.0

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

training_args.bin

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