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.8696 | 5    | 0.6680          | 0.8370   |
+| No log        | 0.8696 | 5    | 0.0346          | 1.0      |
 
 
 ### Framework versions

config.json

@@ -49,20 +49,14 @@
   "hidden_dropout": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "belly_pain",
-    "1": "burping",
-    "2": "discomfort",
-    "3": "hungry",
-    "4": "tired"
+    "0": "no_crying",
+    "1": "crying"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "belly_pain": "0",
-    "burping": "1",
-    "discomfort": "2",
-    "hungry": "3",
-    "tired": "4"
+    "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:1694ae20e5fbe61aaf704f48bc76650982ce9ccee948762667cfe6333c50b2d8
-size 94766588
+oid sha256:95fd422baf1101ad62a2fa7cfa7b7f5cf503a015145e9d9c8a36aea33b9d098e
+size 94763496

runs/Aug13_13-34-29_DesMar/events.out.tfevents.1723548875.DesMar.50772.0

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:93a81a60cb3a42adfe2ffd2035d93813e95993f6b787730e6f8a234b148345b4
+oid sha256:3429db7500655bb90fd39447f09d3edcc319098251582770da5986f307262112
 size 5240