From 670d1e13d8c7a7da0c120b4b7da27c2f2e64fa1d Mon Sep 17 00:00:00 2001 From: Sacha Lewin Date: Fri, 17 Mar 2023 20:54:32 +0100 Subject: [PATCH] Use the run timestamp as run name on WandB (#1619) Makes it way more convenient to run many experiments by splitting by experiment name, and then by timestamp. Co-authored-by: Matthew Tancik --- nerfstudio/utils/writer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/utils/writer.py b/nerfstudio/utils/writer.py index 720d1d449..bffb0bd03 100644 --- a/nerfstudio/utils/writer.py +++ b/nerfstudio/utils/writer.py @@ -283,7 +283,7 @@ class WandbWriter(Writer): """WandDB Writer Class""" def __init__(self, log_dir: Path): - wandb.init(project="nerfstudio-project", dir=str(log_dir), reinit=True) + wandb.init(project="nerfstudio-project", dir=str(log_dir), reinit=True, name=log_dir.name) def write_image(self, name: str, image: TensorType["H", "W", "C"], step: int) -> None: image = torch.permute(image, (2, 0, 1))