Upload 12 files

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:811dc5b8054379df323c2a1f9650afd9bd4e389f9eebfd40a42d916942bf9361
+oid sha256:131da3876aaca1897703a33f04d142f48c945f2450e189fec4a0316b6fce6329
 size 901339568

optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:380a3b6732584bc4ccbd7700ee353b98303cd38b7c9f407278ab8d863e756959
+oid sha256:0fe10f5b77323a7a170ea2031248a93765886fb7c376f51845ee7278381151c1
 size 227854586

rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3458619c66107fd4aa594a8b18101f18110d12fef82506197f9079cfcb95c54b
+oid sha256:4f8e8e8f9d8a69223288f91acadfe178658f954371c9c5e5acf05816f8c72e51
 size 14180

scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:63271abe64cfd32b6594b5297d724e637ca5898f79215f6ae514a34d6714816e
+oid sha256:e773a85e1bc36c2275deb763e1f7c3e5cd88423162895f71855030dbf616e0bb
 size 1064

trainer_state.json

@@ -1,9 +1,9 @@
 {
-  "best_metric": 0.90625,
-  "best_model_checkpoint": "/home01/k157a01/Nguyen-Workdir/bitter/BitT5/output/BTP640_ckpt/sequence_selfies_biot5_plus_base/dropout_0.05/fold_10/checkpoint-2560",
-  "epoch": 40.0,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 50.0,
   "eval_steps": 500,
-  "global_step": 2560,
+  "global_step": 3200,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -22,10 +22,10 @@
       "eval_precision": 0.0,
       "eval_recall": 0.0,
       "eval_roc_auc": 0.630615234375,
-      "eval_runtime": 2.6114,
-      "eval_samples_per_second": 49.016,
+      "eval_runtime": 5.1525,
+      "eval_samples_per_second": 24.843,
       "eval_specificity": 1.0,
-      "eval_steps_per_second": 0.383,
+      "eval_steps_per_second": 0.194,
       "eval_tn": 64,
       "eval_tp": 0,
       "step": 64
@@ -44,10 +44,10 @@
       "eval_precision": 0.6276595744680851,
       "eval_recall": 0.921875,
       "eval_roc_auc": 0.797119140625,
-      "eval_runtime": 2.262,
-      "eval_samples_per_second": 56.588,
+      "eval_runtime": 4.5248,
+      "eval_samples_per_second": 28.288,
       "eval_specificity": 0.453125,
-      "eval_steps_per_second": 0.442,
+      "eval_steps_per_second": 0.221,
       "eval_tn": 29,
       "eval_tp": 59,
       "step": 128
@@ -66,10 +66,10 @@
       "eval_precision": 0.8518518518518519,
       "eval_recall": 0.71875,
       "eval_roc_auc": 0.891845703125,
-      "eval_runtime": 2.2341,
-      "eval_samples_per_second": 57.294,
+      "eval_runtime": 4.5497,
+      "eval_samples_per_second": 28.134,
       "eval_specificity": 0.875,
-      "eval_steps_per_second": 0.448,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 56,
       "eval_tp": 46,
       "step": 192
@@ -88,10 +88,10 @@
       "eval_precision": 0.9090909090909091,
       "eval_recall": 0.78125,
       "eval_roc_auc": 0.934814453125,
-      "eval_runtime": 2.2535,
-      "eval_samples_per_second": 56.801,
+      "eval_runtime": 3.0168,
+      "eval_samples_per_second": 42.43,
       "eval_specificity": 0.921875,
-      "eval_steps_per_second": 0.444,
+      "eval_steps_per_second": 0.331,
       "eval_tn": 59,
       "eval_tp": 50,
       "step": 256
@@ -110,10 +110,10 @@
       "eval_precision": 0.8507462686567164,
       "eval_recall": 0.890625,
       "eval_roc_auc": 0.938232421875,
-      "eval_runtime": 2.2461,
-      "eval_samples_per_second": 56.988,
+      "eval_runtime": 4.3385,
+      "eval_samples_per_second": 29.504,
       "eval_specificity": 0.84375,
-      "eval_steps_per_second": 0.445,
+      "eval_steps_per_second": 0.23,
       "eval_tn": 54,
       "eval_tp": 57,
       "step": 320
@@ -132,10 +132,10 @@
       "eval_precision": 0.8378378378378378,
       "eval_recall": 0.96875,
       "eval_roc_auc": 0.951171875,
-      "eval_runtime": 2.2473,
-      "eval_samples_per_second": 56.957,
+      "eval_runtime": 4.3284,
+      "eval_samples_per_second": 29.572,
       "eval_specificity": 0.8125,
-      "eval_steps_per_second": 0.445,
+      "eval_steps_per_second": 0.231,
       "eval_tn": 52,
       "eval_tp": 62,
       "step": 384
@@ -154,10 +154,10 @@
       "eval_precision": 0.921875,
       "eval_recall": 0.921875,
       "eval_roc_auc": 0.96142578125,
-      "eval_runtime": 2.256,
-      "eval_samples_per_second": 56.738,
+      "eval_runtime": 4.3421,
+      "eval_samples_per_second": 29.479,
       "eval_specificity": 0.921875,
-      "eval_steps_per_second": 0.443,
+      "eval_steps_per_second": 0.23,
       "eval_tn": 59,
       "eval_tp": 59,
       "step": 448
@@ -183,10 +183,10 @@
       "eval_precision": 0.8076923076923077,
       "eval_recall": 0.984375,
       "eval_roc_auc": 0.953369140625,
-      "eval_runtime": 2.2778,
-      "eval_samples_per_second": 56.194,
+      "eval_runtime": 4.3535,
+      "eval_samples_per_second": 29.401,
       "eval_specificity": 0.765625,
-      "eval_steps_per_second": 0.439,
+      "eval_steps_per_second": 0.23,
       "eval_tn": 49,
       "eval_tp": 63,
       "step": 512
@@ -205,10 +205,10 @@
       "eval_precision": 0.8181818181818182,
       "eval_recall": 0.984375,
       "eval_roc_auc": 0.95751953125,
-      "eval_runtime": 2.2828,
-      "eval_samples_per_second": 56.071,
+      "eval_runtime": 4.5389,
+      "eval_samples_per_second": 28.201,
       "eval_specificity": 0.78125,
-      "eval_steps_per_second": 0.438,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 50,
       "eval_tp": 63,
       "step": 576
@@ -227,10 +227,10 @@
       "eval_precision": 0.9322033898305084,
       "eval_recall": 0.859375,
       "eval_roc_auc": 0.9658203125,
-      "eval_runtime": 2.2521,
-      "eval_samples_per_second": 56.837,
+      "eval_runtime": 4.4902,
+      "eval_samples_per_second": 28.507,
       "eval_specificity": 0.9375,
-      "eval_steps_per_second": 0.444,
+      "eval_steps_per_second": 0.223,
       "eval_tn": 60,
       "eval_tp": 55,
       "step": 640
@@ -249,10 +249,10 @@
       "eval_precision": 0.8923076923076924,
       "eval_recall": 0.90625,
       "eval_roc_auc": 0.959716796875,
-      "eval_runtime": 2.2461,
-      "eval_samples_per_second": 56.989,
+      "eval_runtime": 4.487,
+      "eval_samples_per_second": 28.527,
       "eval_specificity": 0.890625,
-      "eval_steps_per_second": 0.445,
+      "eval_steps_per_second": 0.223,
       "eval_tn": 57,
       "eval_tp": 58,
       "step": 704
@@ -271,10 +271,10 @@
       "eval_precision": 0.9333333333333333,
       "eval_recall": 0.875,
       "eval_roc_auc": 0.96240234375,
-      "eval_runtime": 2.2504,
-      "eval_samples_per_second": 56.878,
+      "eval_runtime": 4.5247,
+      "eval_samples_per_second": 28.289,
       "eval_specificity": 0.9375,
-      "eval_steps_per_second": 0.444,
+      "eval_steps_per_second": 0.221,
       "eval_tn": 60,
       "eval_tp": 56,
       "step": 768
@@ -293,10 +293,10 @@
       "eval_precision": 0.9655172413793104,
       "eval_recall": 0.875,
       "eval_roc_auc": 0.963134765625,
-      "eval_runtime": 2.2588,
-      "eval_samples_per_second": 56.668,
+      "eval_runtime": 4.5632,
+      "eval_samples_per_second": 28.051,
       "eval_specificity": 0.96875,
-      "eval_steps_per_second": 0.443,
+      "eval_steps_per_second": 0.219,
       "eval_tn": 62,
       "eval_tp": 56,
       "step": 832
@@ -315,10 +315,10 @@
       "eval_precision": 0.9180327868852459,
       "eval_recall": 0.875,
       "eval_roc_auc": 0.962158203125,
-      "eval_runtime": 2.2545,
-      "eval_samples_per_second": 56.774,
+      "eval_runtime": 4.5111,
+      "eval_samples_per_second": 28.375,
       "eval_specificity": 0.921875,
-      "eval_steps_per_second": 0.444,
+      "eval_steps_per_second": 0.222,
       "eval_tn": 59,
       "eval_tp": 56,
       "step": 896
@@ -337,10 +337,10 @@
       "eval_precision": 0.9180327868852459,
       "eval_recall": 0.875,
       "eval_roc_auc": 0.947509765625,
-      "eval_runtime": 2.2555,
-      "eval_samples_per_second": 56.75,
+      "eval_runtime": 4.5543,
+      "eval_samples_per_second": 28.105,
       "eval_specificity": 0.921875,
-      "eval_steps_per_second": 0.443,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 59,
       "eval_tp": 56,
       "step": 960
@@ -366,10 +366,10 @@
       "eval_precision": 0.9206349206349206,
       "eval_recall": 0.90625,
       "eval_roc_auc": 0.94677734375,
-      "eval_runtime": 2.2802,
-      "eval_samples_per_second": 56.136,
+      "eval_runtime": 4.5536,
+      "eval_samples_per_second": 28.11,
       "eval_specificity": 0.921875,
-      "eval_steps_per_second": 0.439,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 59,
       "eval_tp": 58,
       "step": 1024
@@ -388,10 +388,10 @@
       "eval_precision": 0.9322033898305084,
       "eval_recall": 0.859375,
       "eval_roc_auc": 0.96875,
-      "eval_runtime": 2.2782,
-      "eval_samples_per_second": 56.184,
+      "eval_runtime": 4.548,
+      "eval_samples_per_second": 28.144,
       "eval_specificity": 0.9375,
-      "eval_steps_per_second": 0.439,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 60,
       "eval_tp": 55,
       "step": 1088
@@ -410,10 +410,10 @@
       "eval_precision": 0.9180327868852459,
       "eval_recall": 0.875,
       "eval_roc_auc": 0.95947265625,
-      "eval_runtime": 2.2885,
-      "eval_samples_per_second": 55.933,
+      "eval_runtime": 4.5489,
+      "eval_samples_per_second": 28.139,
       "eval_specificity": 0.921875,
-      "eval_steps_per_second": 0.437,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 59,
       "eval_tp": 56,
       "step": 1152
@@ -432,10 +432,10 @@
       "eval_precision": 0.9230769230769231,
       "eval_recall": 0.9375,
       "eval_roc_auc": 0.958740234375,
-      "eval_runtime": 2.2612,
-      "eval_samples_per_second": 56.607,
+      "eval_runtime": 4.5524,
+      "eval_samples_per_second": 28.117,
       "eval_specificity": 0.921875,
-      "eval_steps_per_second": 0.442,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 59,
       "eval_tp": 60,
       "step": 1216
@@ -454,10 +454,10 @@
       "eval_precision": 0.9104477611940298,
       "eval_recall": 0.953125,
       "eval_roc_auc": 0.951416015625,
-      "eval_runtime": 2.5972,
-      "eval_samples_per_second": 49.284,
+      "eval_runtime": 4.5492,
+      "eval_samples_per_second": 28.137,
       "eval_specificity": 0.90625,
-      "eval_steps_per_second": 0.385,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 58,
       "eval_tp": 61,
       "step": 1280
@@ -476,10 +476,10 @@
       "eval_precision": 0.9491525423728814,
       "eval_recall": 0.875,
       "eval_roc_auc": 0.95654296875,
-      "eval_runtime": 2.285,
-      "eval_samples_per_second": 56.017,
+      "eval_runtime": 4.5431,
+      "eval_samples_per_second": 28.175,
       "eval_specificity": 0.953125,
-      "eval_steps_per_second": 0.438,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 61,
       "eval_tp": 56,
       "step": 1344
@@ -498,10 +498,10 @@
       "eval_precision": 0.9104477611940298,
       "eval_recall": 0.953125,
       "eval_roc_auc": 0.95703125,
-      "eval_runtime": 2.2495,
-      "eval_samples_per_second": 56.902,
+      "eval_runtime": 4.54,
+      "eval_samples_per_second": 28.194,
       "eval_specificity": 0.90625,
-      "eval_steps_per_second": 0.445,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 58,
       "eval_tp": 61,
       "step": 1408
@@ -520,10 +520,10 @@
       "eval_precision": 0.9253731343283582,
       "eval_recall": 0.96875,
       "eval_roc_auc": 0.966796875,
-      "eval_runtime": 2.2552,
-      "eval_samples_per_second": 56.757,
+      "eval_runtime": 4.5302,
+      "eval_samples_per_second": 28.255,
       "eval_specificity": 0.921875,
-      "eval_steps_per_second": 0.443,
+      "eval_steps_per_second": 0.221,
       "eval_tn": 59,
       "eval_tp": 62,
       "step": 1472
@@ -549,10 +549,10 @@
       "eval_precision": 0.9242424242424242,
       "eval_recall": 0.953125,
       "eval_roc_auc": 0.966796875,
-      "eval_runtime": 2.2554,
-      "eval_samples_per_second": 56.753,
+      "eval_runtime": 4.5437,
+      "eval_samples_per_second": 28.171,
       "eval_specificity": 0.921875,
-      "eval_steps_per_second": 0.443,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 59,
       "eval_tp": 61,
       "step": 1536
@@ -571,10 +571,10 @@
       "eval_precision": 0.9242424242424242,
       "eval_recall": 0.953125,
       "eval_roc_auc": 0.964599609375,
-      "eval_runtime": 2.2468,
-      "eval_samples_per_second": 56.97,
+      "eval_runtime": 4.5503,
+      "eval_samples_per_second": 28.13,
       "eval_specificity": 0.921875,
-      "eval_steps_per_second": 0.445,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 59,
       "eval_tp": 61,
       "step": 1600
@@ -593,10 +593,10 @@
       "eval_precision": 0.9508196721311475,
       "eval_recall": 0.90625,
       "eval_roc_auc": 0.958740234375,
-      "eval_runtime": 2.2923,
-      "eval_samples_per_second": 55.84,
+      "eval_runtime": 4.5413,
+      "eval_samples_per_second": 28.186,
       "eval_specificity": 0.953125,
-      "eval_steps_per_second": 0.436,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 61,
       "eval_tp": 58,
       "step": 1664
@@ -615,10 +615,10 @@
       "eval_precision": 0.921875,
       "eval_recall": 0.921875,
       "eval_roc_auc": 0.963134765625,
-      "eval_runtime": 2.2554,
-      "eval_samples_per_second": 56.752,
+      "eval_runtime": 4.5504,
+      "eval_samples_per_second": 28.129,
       "eval_specificity": 0.921875,
-      "eval_steps_per_second": 0.443,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 59,
       "eval_tp": 59,
       "step": 1728
@@ -637,10 +637,10 @@
       "eval_precision": 0.9375,
       "eval_recall": 0.9375,
       "eval_roc_auc": 0.967041015625,
-      "eval_runtime": 2.2678,
-      "eval_samples_per_second": 56.442,
+      "eval_runtime": 4.5474,
+      "eval_samples_per_second": 28.148,
       "eval_specificity": 0.9375,
-      "eval_steps_per_second": 0.441,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 60,
       "eval_tp": 60,
       "step": 1792
@@ -659,10 +659,10 @@
       "eval_precision": 0.9354838709677419,
       "eval_recall": 0.90625,
       "eval_roc_auc": 0.95654296875,
-      "eval_runtime": 2.255,
-      "eval_samples_per_second": 56.763,
+      "eval_runtime": 4.5528,
+      "eval_samples_per_second": 28.115,
       "eval_specificity": 0.9375,
-      "eval_steps_per_second": 0.443,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 60,
       "eval_tp": 58,
       "step": 1856
@@ -681,10 +681,10 @@
       "eval_precision": 0.9516129032258065,
       "eval_recall": 0.921875,
       "eval_roc_auc": 0.96875,
-      "eval_runtime": 2.2628,
-      "eval_samples_per_second": 56.568,
+      "eval_runtime": 4.5887,
+      "eval_samples_per_second": 27.895,
       "eval_specificity": 0.953125,
-      "eval_steps_per_second": 0.442,
+      "eval_steps_per_second": 0.218,
       "eval_tn": 61,
       "eval_tp": 59,
       "step": 1920
@@ -703,10 +703,10 @@
       "eval_precision": 0.9104477611940298,
       "eval_recall": 0.953125,
       "eval_roc_auc": 0.97265625,
-      "eval_runtime": 2.2593,
-      "eval_samples_per_second": 56.655,
+      "eval_runtime": 4.5517,
+      "eval_samples_per_second": 28.121,
       "eval_specificity": 0.90625,
-      "eval_steps_per_second": 0.443,
+      "eval_steps_per_second": 0.22,
       "eval_tn": 58,
       "eval_tp": 61,
       "step": 1984
@@ -732,10 +732,10 @@
       "eval_precision": 0.9375,
       "eval_recall": 0.9375,
       "eval_roc_auc": 0.96875,
-      "eval_runtime": 2.2526,
-      "eval_samples_per_second": 56.823,
+      "eval_runtime": 4.6396,
+      "eval_samples_per_second": 27.588,
       "eval_specificity": 0.9375,
-      "eval_steps_per_second": 0.444,
+      "eval_steps_per_second": 0.216,
       "eval_tn": 60,
       "eval_tp": 60,
       "step": 2048
@@ -754,10 +754,10 @@
       "eval_precision": 0.9242424242424242,
       "eval_recall": 0.953125,
       "eval_roc_auc": 0.966064453125,
-      "eval_runtime": 2.254,
-      "eval_samples_per_second": 56.788,
+      "eval_runtime": 4.676,
+      "eval_samples_per_second": 27.374,
       "eval_specificity": 0.921875,
-      "eval_steps_per_second": 0.444,
+      "eval_steps_per_second": 0.214,
       "eval_tn": 59,
       "eval_tp": 61,
       "step": 2112
@@ -776,10 +776,10 @@
       "eval_precision": 0.9104477611940298,
       "eval_recall": 0.953125,
       "eval_roc_auc": 0.970703125,
-      "eval_runtime": 2.2642,
-      "eval_samples_per_second": 56.532,
+      "eval_runtime": 4.6816,
+      "eval_samples_per_second": 27.341,
       "eval_specificity": 0.90625,
-      "eval_steps_per_second": 0.442,
+      "eval_steps_per_second": 0.214,
       "eval_tn": 58,
       "eval_tp": 61,
       "step": 2176
@@ -798,10 +798,10 @@
       "eval_precision": 0.9242424242424242,
       "eval_recall": 0.953125,
       "eval_roc_auc": 0.963134765625,
-      "eval_runtime": 2.2515,
-      "eval_samples_per_second": 56.851,
+      "eval_runtime": 4.6414,
+      "eval_samples_per_second": 27.578,
       "eval_specificity": 0.921875,
-      "eval_steps_per_second": 0.444,
+      "eval_steps_per_second": 0.215,
       "eval_tn": 59,
       "eval_tp": 61,
       "step": 2240
@@ -820,10 +820,10 @@
       "eval_precision": 0.9516129032258065,
       "eval_recall": 0.921875,
       "eval_roc_auc": 0.96875,
-      "eval_runtime": 2.2593,
-      "eval_samples_per_second": 56.656,
+      "eval_runtime": 4.5146,
+      "eval_samples_per_second": 28.352,
       "eval_specificity": 0.953125,
-      "eval_steps_per_second": 0.443,
+      "eval_steps_per_second": 0.222,
       "eval_tn": 61,
       "eval_tp": 59,
       "step": 2304
@@ -842,10 +842,10 @@
       "eval_precision": 0.9384615384615385,
       "eval_recall": 0.953125,
       "eval_roc_auc": 0.9677734375,
-      "eval_runtime": 2.2945,
-      "eval_samples_per_second": 55.785,
+      "eval_runtime": 4.5189,
+      "eval_samples_per_second": 28.325,
       "eval_specificity": 0.9375,
-      "eval_steps_per_second": 0.436,
+      "eval_steps_per_second": 0.221,
       "eval_tn": 60,
       "eval_tp": 61,
       "step": 2368
@@ -864,10 +864,10 @@
       "eval_precision": 0.9384615384615385,
       "eval_recall": 0.953125,
       "eval_roc_auc": 0.961181640625,
-      "eval_runtime": 2.2598,
-      "eval_samples_per_second": 56.642,
+      "eval_runtime": 4.6226,
+      "eval_samples_per_second": 27.69,
       "eval_specificity": 0.9375,
-      "eval_steps_per_second": 0.443,
+      "eval_steps_per_second": 0.216,
       "eval_tn": 60,
       "eval_tp": 61,
       "step": 2432
@@ -886,10 +886,10 @@
       "eval_precision": 0.9384615384615385,
       "eval_recall": 0.953125,
       "eval_roc_auc": 0.96435546875,
-      "eval_runtime": 2.2495,
-      "eval_samples_per_second": 56.901,
+      "eval_runtime": 4.5344,
+      "eval_samples_per_second": 28.229,
       "eval_specificity": 0.9375,
-      "eval_steps_per_second": 0.445,
+      "eval_steps_per_second": 0.221,
       "eval_tn": 60,
       "eval_tp": 61,
       "step": 2496
@@ -915,13 +915,240 @@
       "eval_precision": 0.953125,
       "eval_recall": 0.953125,
       "eval_roc_auc": 0.96240234375,
-      "eval_runtime": 2.2796,
-      "eval_samples_per_second": 56.149,
+      "eval_runtime": 4.5169,
+      "eval_samples_per_second": 28.338,
       "eval_specificity": 0.953125,
-      "eval_steps_per_second": 0.439,
+      "eval_steps_per_second": 0.221,
       "eval_tn": 61,
       "eval_tp": 61,
       "step": 2560
+    },
+    {
+      "epoch": 41.0,
+      "eval_accuracy": 0.9296875,
+      "eval_auc": 0.96435546875,
+      "eval_balanced_acc": 0.9296875,
+      "eval_f1": 0.928,
+      "eval_fn": 6,
+      "eval_fp": 3,
+      "eval_gmean": 0.9293920223726907,
+      "eval_loss": 0.23136085271835327,
+      "eval_matthews_correlation": 0.8603206963184278,
+      "eval_precision": 0.9508196721311475,
+      "eval_recall": 0.90625,
+      "eval_roc_auc": 0.96435546875,
+      "eval_runtime": 4.5098,
+      "eval_samples_per_second": 28.383,
+      "eval_specificity": 0.953125,
+      "eval_steps_per_second": 0.222,
+      "eval_tn": 61,
+      "eval_tp": 58,
+      "step": 2624
+    },
+    {
+      "epoch": 42.0,
+      "eval_accuracy": 0.9453125,
+      "eval_auc": 0.9619140625,
+      "eval_balanced_acc": 0.9453125,
+      "eval_f1": 0.9457364341085271,
+      "eval_fn": 3,
+      "eval_fp": 4,
+      "eval_gmean": 0.9452802163908859,
+      "eval_loss": 0.22938130795955658,
+      "eval_matthews_correlation": 0.8907337387831413,
+      "eval_precision": 0.9384615384615385,
+      "eval_recall": 0.953125,
+      "eval_roc_auc": 0.9619140625,
+      "eval_runtime": 4.5043,
+      "eval_samples_per_second": 28.418,
+      "eval_specificity": 0.9375,
+      "eval_steps_per_second": 0.222,
+      "eval_tn": 60,
+      "eval_tp": 61,
+      "step": 2688
+    },
+    {
+      "epoch": 43.0,
+      "eval_accuracy": 0.9375,
+      "eval_auc": 0.955810546875,
+      "eval_balanced_acc": 0.9375,
+      "eval_f1": 0.9365079365079365,
+      "eval_fn": 5,
+      "eval_fp": 3,
+      "eval_gmean": 0.9373697826231652,
+      "eval_loss": 0.24056339263916016,
+      "eval_matthews_correlation": 0.8754275592730114,
+      "eval_precision": 0.9516129032258065,
+      "eval_recall": 0.921875,
+      "eval_roc_auc": 0.955810546875,
+      "eval_runtime": 4.5092,
+      "eval_samples_per_second": 28.387,
+      "eval_specificity": 0.953125,
+      "eval_steps_per_second": 0.222,
+      "eval_tn": 61,
+      "eval_tp": 59,
+      "step": 2752
+    },
+    {
+      "epoch": 44.0,
+      "eval_accuracy": 0.953125,
+      "eval_auc": 0.957763671875,
+      "eval_balanced_acc": 0.953125,
+      "eval_f1": 0.953125,
+      "eval_fn": 3,
+      "eval_fp": 3,
+      "eval_gmean": 0.953125,
+      "eval_loss": 0.22141942381858826,
+      "eval_matthews_correlation": 0.90625,
+      "eval_precision": 0.953125,
+      "eval_recall": 0.953125,
+      "eval_roc_auc": 0.957763671875,
+      "eval_runtime": 4.5337,
+      "eval_samples_per_second": 28.233,
+      "eval_specificity": 0.953125,
+      "eval_steps_per_second": 0.221,
+      "eval_tn": 61,
+      "eval_tp": 61,
+      "step": 2816
+    },
+    {
+      "epoch": 45.0,
+      "eval_accuracy": 0.953125,
+      "eval_auc": 0.961669921875,
+      "eval_balanced_acc": 0.953125,
+      "eval_f1": 0.953125,
+      "eval_fn": 3,
+      "eval_fp": 3,
+      "eval_gmean": 0.953125,
+      "eval_loss": 0.22277112305164337,
+      "eval_matthews_correlation": 0.90625,
+      "eval_precision": 0.953125,
+      "eval_recall": 0.953125,
+      "eval_roc_auc": 0.961669921875,
+      "eval_runtime": 2.3022,
+      "eval_samples_per_second": 55.599,
+      "eval_specificity": 0.953125,
+      "eval_steps_per_second": 0.434,
+      "eval_tn": 61,
+      "eval_tp": 61,
+      "step": 2880
+    },
+    {
+      "epoch": 46.0,
+      "eval_accuracy": 0.953125,
+      "eval_auc": 0.95458984375,
+      "eval_balanced_acc": 0.953125,
+      "eval_f1": 0.953125,
+      "eval_fn": 3,
+      "eval_fp": 3,
+      "eval_gmean": 0.953125,
+      "eval_loss": 0.2253548949956894,
+      "eval_matthews_correlation": 0.90625,
+      "eval_precision": 0.953125,
+      "eval_recall": 0.953125,
+      "eval_roc_auc": 0.95458984375,
+      "eval_runtime": 4.5066,
+      "eval_samples_per_second": 28.403,
+      "eval_specificity": 0.953125,
+      "eval_steps_per_second": 0.222,
+      "eval_tn": 61,
+      "eval_tp": 61,
+      "step": 2944
+    },
+    {
+      "epoch": 46.875,
+      "grad_norm": 0.10018020868301392,
+      "learning_rate": 3.289473684210526e-05,
+      "loss": 0.0659,
+      "step": 3000
+    },
+    {
+      "epoch": 47.0,
+      "eval_accuracy": 0.953125,
+      "eval_auc": 0.96337890625,
+      "eval_balanced_acc": 0.953125,
+      "eval_f1": 0.953125,
+      "eval_fn": 3,
+      "eval_fp": 3,
+      "eval_gmean": 0.953125,
+      "eval_loss": 0.22502213716506958,
+      "eval_matthews_correlation": 0.90625,
+      "eval_precision": 0.953125,
+      "eval_recall": 0.953125,
+      "eval_roc_auc": 0.96337890625,
+      "eval_runtime": 4.5194,
+      "eval_samples_per_second": 28.322,
+      "eval_specificity": 0.953125,
+      "eval_steps_per_second": 0.221,
+      "eval_tn": 61,
+      "eval_tp": 61,
+      "step": 3008
+    },
+    {
+      "epoch": 48.0,
+      "eval_accuracy": 0.953125,
+      "eval_auc": 0.963623046875,
+      "eval_balanced_acc": 0.953125,
+      "eval_f1": 0.953125,
+      "eval_fn": 3,
+      "eval_fp": 3,
+      "eval_gmean": 0.953125,
+      "eval_loss": 0.2254004329442978,
+      "eval_matthews_correlation": 0.90625,
+      "eval_precision": 0.953125,
+      "eval_recall": 0.953125,
+      "eval_roc_auc": 0.963623046875,
+      "eval_runtime": 4.5174,
+      "eval_samples_per_second": 28.335,
+      "eval_specificity": 0.953125,
+      "eval_steps_per_second": 0.221,
+      "eval_tn": 61,
+      "eval_tp": 61,
+      "step": 3072
+    },
+    {
+      "epoch": 49.0,
+      "eval_accuracy": 0.953125,
+      "eval_auc": 0.96826171875,
+      "eval_balanced_acc": 0.953125,
+      "eval_f1": 0.953125,
+      "eval_fn": 3,
+      "eval_fp": 3,
+      "eval_gmean": 0.953125,
+      "eval_loss": 0.2249780297279358,
+      "eval_matthews_correlation": 0.90625,
+      "eval_precision": 0.953125,
+      "eval_recall": 0.953125,
+      "eval_roc_auc": 0.96826171875,
+      "eval_runtime": 4.5274,
+      "eval_samples_per_second": 28.272,
+      "eval_specificity": 0.953125,
+      "eval_steps_per_second": 0.221,
+      "eval_tn": 61,
+      "eval_tp": 61,
+      "step": 3136
+    },
+    {
+      "epoch": 50.0,
+      "eval_accuracy": 0.953125,
+      "eval_auc": 0.9677734375,
+      "eval_balanced_acc": 0.953125,
+      "eval_f1": 0.953125,
+      "eval_fn": 3,
+      "eval_fp": 3,
+      "eval_gmean": 0.953125,
+      "eval_loss": 0.2252984642982483,
+      "eval_matthews_correlation": 0.90625,
+      "eval_precision": 0.953125,
+      "eval_recall": 0.953125,
+      "eval_roc_auc": 0.9677734375,
+      "eval_runtime": 4.5509,
+      "eval_samples_per_second": 28.127,
+      "eval_specificity": 0.953125,
+      "eval_steps_per_second": 0.22,
+      "eval_tn": 61,
+      "eval_tp": 61,
+      "step": 3200
     }
   ],
   "logging_steps": 500,
@@ -936,12 +1163,12 @@
         "should_evaluate": false,
         "should_log": false,
         "should_save": true,
-        "should_training_stop": false
+        "should_training_stop": true
       },
       "attributes": {}
     }
   },
-  "total_flos": 4941874692922944.0,
+  "total_flos": 6180004916191008.0,
   "train_batch_size": 8,
   "trial_name": null,
   "trial_params": null

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1516a1fc7553b75ab4a9d3a6428d3d16a96ea0ab2f170c13a7557b24e0afc49c
-size 5560
+oid sha256:0836c1d1011341b50b6969f934d0a13595b9d18eb2158049c63cbba331fef33a
+size 5496