Fix ns-train help (#1370)

This commit is contained in:
Jonáš Kulhánek authored and GitHub committed 2023-02-07 09:49:09 -08:00
1 parent e749fbcf78
commit 6136745d75
1 file changed
+2 -1
+2 -1
View File
@@ -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: