Add training finished statement (#914)

nerfstudio-project · Nov 7, 2022 · 49e8936 · 49e8936
1 parent 39462e7
commit 49e8936
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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