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

config.json

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

runs/Aug13_14-16-16_DesMar/events.out.tfevents.1723551378.DesMar.79020.0

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

training_args.bin

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