tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Code

Update optimizer docstrings (#1333)

Browse Source
This commit is contained in:
Matthew Tancik 2023-02-01 15:15:08 -08:00 committed by GitHub
parent 48ec36e53f
commit db83068fcd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nerfstudio/engine/optimizers.py
View File

@ -34,9 +34,13 @@ class OptimizerConfig(base_config.PrintableConfig):
"""Basic optimizer config with RAdam"""
_target: Type = torch.optim.Adam
"""The optimizer class to use."""
lr: float = 0.0005
"""The learning rate to use."""
eps: float = 1e-08
"""The epsilon value to use."""
max_norm: Optional[float] = None
"""The max norm to use for gradient clipping."""
# TODO: somehow make this more generic. i dont like the idea of overriding the setup function
# but also not sure how to go about passing things into predefined torch objects.
@ -54,6 +58,7 @@ class AdamOptimizerConfig(OptimizerConfig):
_target: Type = torch.optim.Adam
weight_decay: float = 0
"""The weight decay to use."""
@dataclass