fix: missing val_bpb on resume

This commit is contained in:
Sanzo00 2025-11-22 11:04:20 +08:00
parent 4a87a0d19f
commit 53b3a4fb81

View File

@ -205,6 +205,7 @@ if not resuming:
else: else:
step = meta_data["step"] step = meta_data["step"]
loop_state = meta_data["loop_state"] loop_state = meta_data["loop_state"]
val_bpb = meta_data["val_bpb"]
min_val_bpb = loop_state["min_val_bpb"] min_val_bpb = loop_state["min_val_bpb"]
smooth_train_loss = loop_state["smooth_train_loss"] smooth_train_loss = loop_state["smooth_train_loss"]
total_training_time = loop_state["total_training_time"] total_training_time = loop_state["total_training_time"]