mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-12 19:04:03 +08:00
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 <tancik@berkeley.edu>
This commit is contained in:
1 parent
505ebe6793
commit
670d1e13d8
1 file changed
+1
-1
@@ -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))
|
||||
|
||||
Reference in new issue
Block a user