Checkpoint at epoch 6

epoch_6/metadata.json

@@ -1,21 +1,21 @@
 {
   "epoch": 6,
-  "loss": 0.02499583659702888,
+  "loss": 0.028372820999877566,
   "results": {
-    "loss": 0.02499583659702888,
-    "accuracy": 0.926702624376107,
-    "balanced_accuracy": 0.9570773970762437,
-    "auroc": 0.9960873933079069,
-    "auprc": 0.9993633957853147,
-    "TP": 20487,
-    "TN": 2536,
-    "FP": 12,
-    "FN": 1809,
-    "precision_pos": 0.9994146055905166,
-    "recall_pos": 0.9188643702906351,
-    "f1_pos": 0.9574483000350508,
-    "precision_neg": 0.5836593785960874,
-    "recall_neg": 0.9952904238618524,
-    "f1_neg": 0.7358189467575801
+    "loss": 0.028372820999877566,
+    "accuracy": 0.9237642891643858,
+    "balanced_accuracy": 0.9542236845885161,
+    "auroc": 0.9956083244757703,
+    "auprc": 0.9993203523992252,
+    "TP": 20421,
+    "TN": 2529,
+    "FP": 19,
+    "FN": 1875,
+    "precision_pos": 0.9990704500978473,
+    "recall_pos": 0.9159041980624327,
+    "f1_pos": 0.9556813927368027,
+    "precision_neg": 0.5742506811989101,
+    "recall_neg": 0.9925431711145997,
+    "f1_neg": 0.7275604142692751
   }
 }

epoch_6/model_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:19fc50c3ae0ee1906df8588302b98f07835037fc978299b4ffec27aeb3fbc002
+oid sha256:55dc138fc4cf4057591c6eb8e8f6d9e47e844c8739ae1b12b39d8064d3d31644
 size 377987876

epoch_6/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:41f60d0dc291b56d5f55e594170bc71928b6f547f2e132fb4641ae82f171d521
+oid sha256:4422d805131e777fa6f27de2d886e004605699516549e2e59417a64e8ae2f009
 size 827237