From c6692a25f9c808f6bb12b87d7fdf3fb638e32dfb Mon Sep 17 00:00:00 2001
From: Matt Tancik <tancik@berkeley.edu>
Date: Sun, 6 Nov 2022 22:54:39 -0800
Subject: [PATCH] Add training finished statement

---
 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