Checkpoint at epoch 16

epoch_16/metadata.json

@@ -1,21 +1,21 @@
 {
   "epoch": 16,
-  "loss": 0.06981658024799248,
+  "loss": 0.020973448980122218,
   "results": {
-    "loss": 0.06981658024799248,
-    "accuracy": 0.959950088552568,
-    "balanced_accuracy": 0.850633815669184,
-    "auroc": 0.9836908887923805,
-    "auprc": 0.9979583399436031,
-    "TP": 22032,
-    "TN": 1817,
-    "FP": 731,
-    "FN": 264,
-    "precision_pos": 0.9678864824495892,
-    "recall_pos": 0.9881593110871906,
-    "f1_pos": 0.9779178410528417,
-    "precision_neg": 0.8731379144641999,
-    "recall_neg": 0.7131083202511774,
-    "f1_neg": 0.7850507669042989
+    "loss": 0.020973448980122218,
+    "accuracy": 0.9877636451457092,
+    "balanced_accuracy": 0.9700663303327458,
+    "auroc": 0.9962926557142688,
+    "auprc": 0.9993684807636956,
+    "TP": 22125,
+    "TN": 2415,
+    "FP": 133,
+    "FN": 171,
+    "precision_pos": 0.9940246203612184,
+    "recall_pos": 0.9923304628632938,
+    "f1_pos": 0.9931768191408178,
+    "precision_neg": 0.9338747099767981,
+    "recall_neg": 0.9478021978021978,
+    "f1_neg": 0.9407869107908065
   }
 }

epoch_16/model_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:876491616278276ffef720d680ec10033860bf2412494ce67145bbd61351a3d9
+oid sha256:8f13af0015114d2751876b6c3c178e90700c6ad739ddb826ee46bd67bfad38b6
 size 377988324

epoch_16/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:144b2de74d9c8b57971346d710bc234b3dee2bb737ea35c3115ba616df943e72
+oid sha256:9edef62f2473091636306f5e8cedce2740c58bdf4ac8baa6e88cb2c9be0eb4e3
 size 827237