stage2: Epoch 0/10 - Latest checkpoint

statistics.json

@@ -17,31 +17,17 @@
   "language_4bit_quantized": false,
   "memory_158bit_quantized": true,
   "estimated_model_size_mb": 1026.85,
-  "epoch": 2,
+  "epoch": 0,
   "total_epochs": 10,
   "stage": "stage2",
-  "timestamp": "2025-12-19T07:42:32.263529",
+  "timestamp": "2025-12-19T21:35:56.376020",
   "training_history": [
     {
       "epoch": 0,
-      "train_loss": 1.712057786896447,
+      "train_loss": 1.7373690557870052,
       "alignment_loss": 0,
       "learning_rate": 9.786403637412737e-05,
       "gradient_norm": 0
-    },
-    {
-      "epoch": 1,
-      "train_loss": 1.4130540664910034,
-      "alignment_loss": 0,
-      "learning_rate": 9.098753048166504e-05,
-      "gradient_norm": 0
-    },
-    {
-      "epoch": 2,
-      "train_loss": 1.353085695717397,
-      "alignment_loss": 0,
-      "learning_rate": 8.00400991267938e-05,
-      "gradient_norm": 0
     }
   ]
 }