From 53b3a4fb814d845107ed2a43db97cd2232433c68 Mon Sep 17 00:00:00 2001 From: Sanzo00 Date: Sat, 22 Nov 2025 11:04:20 +0800 Subject: [PATCH] fix: missing val_bpb on resume --- scripts/base_train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/scripts/base_train.py b/scripts/base_train.py index c9ea6c9..72ee147 100644 --- a/scripts/base_train.py +++ b/scripts/base_train.py @@ -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"]