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

Allow the option for timestamp to be generated for ns-train --load-config (#2442)

Browse Source 
* Fix timestamp for ns-train --load-config

* Remove whitespace
This commit is contained in:
フラハティ　仁 2023-09-19 13:13:48 +09:00 committed by GitHub
parent ef2fd3dbe1
commit b0e700137b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nerfstudio/scripts/train.py
View File

@ -226,7 +226,6 @@ def launch(
def main(config: TrainerConfig) -> None:
"""Main function."""
config.set_timestamp()
if config.data:
CONSOLE.log("Using --data alias for --data.pipeline.datamanager.data")
config.pipeline.datamanager.data = config.data
@ -239,6 +238,8 @@ def main(config: TrainerConfig) -> None:
CONSOLE.log(f"Loading pre-set config from: {config.load_config}")
config = yaml.load(config.load_config.read_text(), Loader=yaml.Loader)
config.set_timestamp()
# print and save config
config.print_to_terminal()
config.save_config()