ft-wav2vec2-with-minds

README.md

@@ -15,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model was trained from scratch on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0119
+- Loss: 0.0333
 - Accuracy: 0.9972
 
 ## Model description
@@ -44,63 +44,23 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 50
+- num_epochs: 10
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 5.1595        | 1.0   | 9    | 5.2504          | 0.1125   |
-| 3.706         | 2.0   | 18   | 1.7425          | 0.3261   |
-| 1.1096        | 3.0   | 27   | 0.5152          | 0.7985   |
-| 0.3567        | 4.0   | 36   | 0.1222          | 0.9728   |
-| 0.1645        | 5.0   | 45   | 0.0988          | 0.9850   |
-| 0.1539        | 6.0   | 54   | 0.0696          | 0.9878   |
-| 0.1329        | 7.0   | 63   | 0.0783          | 0.9831   |
-| 0.1023        | 8.0   | 72   | 0.0833          | 0.9841   |
-| 0.1923        | 9.0   | 81   | 0.0733          | 0.9775   |
-| 0.108         | 10.0  | 90   | 0.0294          | 0.9934   |
-| 0.0884        | 11.0  | 99   | 0.0331          | 0.9897   |
-| 0.1745        | 12.0  | 108  | 0.0288          | 0.9944   |
-| 0.0793        | 13.0  | 117  | 0.0545          | 0.9869   |
-| 0.0823        | 14.0  | 126  | 0.0551          | 0.9850   |
-| 0.0857        | 15.0  | 135  | 0.0401          | 0.9925   |
-| 0.0738        | 16.0  | 144  | 0.0329          | 0.9906   |
-| 0.0905        | 17.0  | 153  | 0.0324          | 0.9878   |
-| 0.1049        | 18.0  | 162  | 0.0379          | 0.9925   |
-| 0.0775        | 19.0  | 171  | 0.0410          | 0.9906   |
-| 0.07          | 20.0  | 180  | 0.0315          | 0.9925   |
-| 0.0519        | 21.0  | 189  | 0.0361          | 0.9897   |
-| 0.0679        | 22.0  | 198  | 0.0470          | 0.9878   |
-| 0.0771        | 23.0  | 207  | 0.0258          | 0.9934   |
-| 0.0588        | 24.0  | 216  | 0.0322          | 0.9934   |
-| 0.0566        | 25.0  | 225  | 0.0251          | 0.9906   |
-| 0.0665        | 26.0  | 234  | 0.0162          | 0.9963   |
-| 0.06          | 27.0  | 243  | 0.0178          | 0.9953   |
-| 0.0462        | 28.0  | 252  | 0.0183          | 0.9944   |
-| 0.0527        | 29.0  | 261  | 0.0669          | 0.9831   |
-| 0.0378        | 30.0  | 270  | 0.0163          | 0.9953   |
-| 0.0418        | 31.0  | 279  | 0.0207          | 0.9963   |
-| 0.0335        | 32.0  | 288  | 0.0159          | 0.9953   |
-| 0.0447        | 33.0  | 297  | 0.0151          | 0.9963   |
-| 0.0455        | 34.0  | 306  | 0.0161          | 0.9953   |
-| 0.0368        | 35.0  | 315  | 0.0163          | 0.9944   |
-| 0.043         | 36.0  | 324  | 0.0136          | 0.9963   |
-| 0.0361        | 37.0  | 333  | 0.0181          | 0.9963   |
-| 0.0374        | 38.0  | 342  | 0.0149          | 0.9963   |
-| 0.0397        | 39.0  | 351  | 0.0119          | 0.9963   |
-| 0.0329        | 40.0  | 360  | 0.0164          | 0.9953   |
-| 0.0933        | 41.0  | 369  | 0.0119          | 0.9972   |
-| 0.0311        | 42.0  | 378  | 0.0144          | 0.9963   |
-| 0.0325        | 43.0  | 387  | 0.0131          | 0.9963   |
-| 0.0418        | 44.0  | 396  | 0.0207          | 0.9963   |
-| 0.0251        | 45.0  | 405  | 0.0178          | 0.9963   |
-| 0.0409        | 46.0  | 414  | 0.0149          | 0.9953   |
-| 0.0444        | 47.0  | 423  | 0.0155          | 0.9953   |
-| 0.0318        | 48.0  | 432  | 0.0169          | 0.9953   |
-| 0.0465        | 49.0  | 441  | 0.0171          | 0.9953   |
-| 0.0308        | 50.0  | 450  | 0.0173          | 0.9953   |
+| 4.6092        | 1.0   | 9    | 2.4860          | 0.4311   |
+| 1.4641        | 2.0   | 18   | 0.5758          | 0.7826   |
+| 0.5061        | 3.0   | 27   | 0.1966          | 0.9756   |
+| 0.2573        | 4.0   | 36   | 0.1038          | 0.9803   |
+| 0.1557        | 5.0   | 45   | 0.0671          | 0.9859   |
+| 0.1235        | 6.0   | 54   | 0.0333          | 0.9972   |
+| 0.0725        | 7.0   | 63   | 0.0334          | 0.9944   |
+| 0.0914        | 8.0   | 72   | 0.0279          | 0.9953   |
+| 0.1695        | 9.0   | 81   | 0.0276          | 0.9972   |
+| 0.1118        | 10.0  | 90   | 0.0290          | 0.9972   |
 
 
 ### Framework versions

config.json

@@ -59,26 +59,26 @@
   "hidden_dropout_prob": 0.1,
   "hidden_size": 1024,
   "id2label": {
-    "0": "yes",
+    "0": "stop",
     "1": "right",
-    "2": "up",
-    "3": "stop",
-    "4": "down",
-    "5": "no",
-    "6": "left",
+    "2": "no",
+    "3": "up",
+    "4": "yes",
+    "5": "left",
+    "6": "down",
     "7": "go"
   },
   "initializer_range": 0.02,
   "intermediate_size": 4096,
   "label2id": {
-    "down": "4",
+    "down": "6",
     "go": "7",
-    "left": "6",
-    "no": "5",
+    "left": "5",
+    "no": "2",
     "right": "1",
-    "stop": "3",
-    "up": "2",
-    "yes": "0"
+    "stop": "0",
+    "up": "3",
+    "yes": "4"
   },
   "layer_norm_eps": 1e-05,
   "layerdrop": 0.1,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:35a3a121ad49ab1e260e6c6ffecbc68964596cfd9db970975fe61420fd6df36f
+oid sha256:3e062be6c6ede8fb4afd5974779ebc3d0fb992f8c30ffbf774ec4051ab83fcd7
 size 1262824104

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:02d1047b983cef30dadc280a3634830cf4d2704ec0ce70d1261b8be6b9e7ea75
+oid sha256:f32cd55c10b77470125a0af5d492752771bf7144f0839051de08de47a3b951d6
 size 4143