fix: missing val_bpb on resume

This commit is contained in:
Andrej 2025-12-08 12:35:08 -08:00 committed by GitHub
commit 2fd0440355
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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"]