diff --git a/nerfstudio/engine/optimizers.py b/nerfstudio/engine/optimizers.py index d2cdb022a..8a2a063b2 100644 --- a/nerfstudio/engine/optimizers.py +++ b/nerfstudio/engine/optimizers.py @@ -66,6 +66,8 @@ class RAdamOptimizerConfig(OptimizerConfig): """Basic optimizer config with RAdam""" _target: Type = torch.optim.RAdam + weight_decay: float = 0 + """The weight decay to use.""" class Optimizers: