From 49e89365bc09d166e6f6018f81dc3cd2b8b91927 Mon Sep 17 00:00:00 2001
From: Matthew Tancik <tancik@berkeley.edu>
Date: Sun, 6 Nov 2022 22:55:50 -0800
Subject: [PATCH] Add training finished statement (#914)

---
 nerfstudio/engine/trainer.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/nerfstudio/engine/trainer.py b/nerfstudio/engine/trainer.py
index 74e3d25bf8..5b5bca84c9 100644
--- a/nerfstudio/engine/trainer.py
+++ b/nerfstudio/engine/trainer.py
@@ -174,6 +174,9 @@ def train(self) -> None:
                 writer.write_out_storage()
             # save checkpoint at the end of training
             self.save_checkpoint(step)
+            CONSOLE.rule()
+            CONSOLE.print("[bold green]:tada: :tada: :tada: Training Finished :tada: :tada: :tada:", justify="center")
+            CONSOLE.print("Use ctrl+c to quit", justify="center")
             self._always_render(step)
 
     @check_main_thread