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