From 6136745d75b484bb2306fb99ef39d966daaf6df6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jon=C3=A1=C5=A1=20Kulh=C3=A1nek?= Date: Tue, 7 Feb 2023 18:49:09 +0100 Subject: [PATCH] Fix ns-train help (#1370) --- nerfstudio/engine/trainer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nerfstudio/engine/trainer.py b/nerfstudio/engine/trainer.py index 2d4a45395..d0e750c64 100644 --- a/nerfstudio/engine/trainer.py +++ b/nerfstudio/engine/trainer.py @@ -77,8 +77,9 @@ class TrainerConfig(ExperimentConfig): load_step: Optional[int] = None """Optionally specify model step to load from; if none, will find most recent model in load_dir.""" load_config: Optional[Path] = None - """Optionally log gradients during training""" + """Path to config YAML file.""" log_gradients: bool = False + """Optionally log gradients during training""" class Trainer: