From b0e700137bd8bde21fbc7867f433dd36b2905e1e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E3=83=95=E3=83=A9=E3=83=8F=E3=83=86=E3=82=A3=E3=80=80?= =?UTF-8?q?=E4=BB=81?= Date: Tue, 19 Sep 2023 13:13:48 +0900 Subject: [PATCH] Allow the option for timestamp to be generated for `ns-train --load-config` (#2442) * Fix timestamp for ns-train --load-config * Remove whitespace --- nerfstudio/scripts/train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nerfstudio/scripts/train.py b/nerfstudio/scripts/train.py index 9ca18c71a..fde0a5c37 100644 --- a/nerfstudio/scripts/train.py +++ b/nerfstudio/scripts/train.py @@ -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()