diff --git a/train.py b/train.py index 2e743974..f7bf1450 100644 --- a/train.py +++ b/train.py @@ -620,6 +620,7 @@ def get_weight_decay(progress): print("---") print(f"val_bpb: {val_bpb:.6f}") +print(f"startup_seconds: {startup_time:.1f}") print(f"training_seconds: {total_training_time:.1f}") print(f"total_seconds: {t_end - t_start:.1f}") print(f"peak_vram_mb: {peak_vram_mb:.1f}")