diff --git a/nerfstudio/engine/optimizers.py b/nerfstudio/engine/optimizers.py index 3e147dcdc..b80b77bcb 100644 --- a/nerfstudio/engine/optimizers.py +++ b/nerfstudio/engine/optimizers.py @@ -108,7 +108,7 @@ class Optimizers: Args: param_group_name: name of scheduler to step forward """ - if self.config.param_group_name.scheduler: # type: ignore + if "scheduler" in self.config[param_group_name]: self.schedulers[param_group_name].step() def zero_grad_all(self) -> None: