Checkpoint at epoch 10

epoch_10/metadata.json

@@ -1,21 +1,21 @@
 {
   "epoch": 10,
-  "loss": 0.03922487342187193,
+  "loss": 0.021367146059279803,
   "results": {
-    "loss": 0.03922487342187193,
-    "accuracy": 0.9265416197069715,
-    "balanced_accuracy": 0.9217049161305658,
-    "auroc": 0.9774148160133476,
-    "auprc": 0.9970622472578972,
-    "TP": 20686,
-    "TN": 2333,
-    "FP": 215,
-    "FN": 1610,
-    "precision_pos": 0.9897134108415865,
-    "recall_pos": 0.9277897380696088,
-    "f1_pos": 0.9577516957196102,
-    "precision_neg": 0.5916814608166371,
-    "recall_neg": 0.9156200941915228,
-    "f1_neg": 0.7188414728085041
+    "loss": 0.021367146059279803,
+    "accuracy": 0.9849058122685558,
+    "balanced_accuracy": 0.9663884349798544,
+    "auroc": 0.9956058337262205,
+    "auprc": 0.9993203985738164,
+    "TP": 22066,
+    "TN": 2403,
+    "FP": 145,
+    "FN": 230,
+    "precision_pos": 0.9934717032101211,
+    "recall_pos": 0.9896842482956584,
+    "f1_pos": 0.9915743590895815,
+    "precision_neg": 0.9126471705279149,
+    "recall_neg": 0.9430926216640503,
+    "f1_neg": 0.9276201505500868
   }
 }

epoch_10/model_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4c0a6c8edccb0f2dd4106ef499708c848420819a8bc20087e193b36c3d45db0d
+oid sha256:4bb7192f1275f3eac8a7aa0dbec6147ff14f73744ae76b34d978a24b75cac8d0
 size 377988324

epoch_10/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ef8630bec0d7093e89280c2d96c67fa861ddd124627105994adfdfdebdce2a3e
+oid sha256:5f1d62a7ed368922b4ffe1abc3cc52d3d0c9246ad39394c509dfbc541b5b9257
 size 827237