tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 18:44:21 +08:00
Code

Add training finished statement (#914)

Browse Source
This commit is contained in:
Matthew Tancik 2022-11-06 22:55:50 -08:00 committed by GitHub
parent 39462e78b5
commit 49e89365bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nerfstudio/engine/trainer.py
View File

@ -174,6 +174,9 @@ class Trainer:
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