fix: missing val_bpb on resume

2025-12-08 12:35:08 -08:00
parent 4a87a0d19f 53b3a4fb81
commit 2fd0440355
1 changed files with 1 additions and 0 deletions
@@ -205,6 +205,7 @@ if not resuming:
 else:
    step = meta_data["step"]
    loop_state = meta_data["loop_state"]
+    val_bpb = meta_data["val_bpb"]
    min_val_bpb = loop_state["min_val_bpb"]
    smooth_train_loss = loop_state["smooth_train_loss"]
    total_training_time = loop_state["total_training_time"]