Checkpoint at epoch 2

epoch_2/metadata.json

@@ -1,21 +1,21 @@
 {
   "epoch": 2,
-  "loss": 0.06416799726163636,
+  "loss": 0.032588113567095484,
   "results": {
-    "loss": 0.06416799726163636,
-    "accuracy": 0.8896312993076799,
-    "balanced_accuracy": 0.8750696705775132,
-    "auroc": 0.9440257374167684,
-    "auprc": 0.9922782007380578,
-    "TP": 19919,
-    "TN": 2183,
-    "FP": 365,
-    "FN": 2377,
-    "precision_pos": 0.9820055215933741,
-    "recall_pos": 0.893388948690348,
-    "f1_pos": 0.9356035697510569,
-    "precision_neg": 0.4787280701754386,
-    "recall_neg": 0.8567503924646782,
-    "f1_neg": 0.6142374788970175
+    "loss": 0.032588113567095484,
+    "accuracy": 0.9546771856383836,
+    "balanced_accuracy": 0.9401612822822265,
+    "auroc": 0.9869930594163641,
+    "auprc": 0.9984321754982379,
+    "TP": 21369,
+    "TN": 2349,
+    "FP": 199,
+    "FN": 927,
+    "precision_pos": 0.9907733679525222,
+    "recall_pos": 0.9584230355220668,
+    "f1_pos": 0.9743297464891483,
+    "precision_neg": 0.717032967032967,
+    "recall_neg": 0.9218995290423861,
+    "f1_neg": 0.8066620879120878
   }
 }

epoch_2/model_state_dict.pth

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

epoch_2/optimizer_state_dict.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:315a752f7e4eb863c90c9e3104015f6c8faec29c3940248da905551ddf2c358e
+oid sha256:89f6c67913dbd942ee796ee868070624cc650c44b7c047beeee6c31919a6e434
 size 827237