From cba8d22d3bed8b1458ec552195aef5c28c7cee97 Mon Sep 17 00:00:00 2001 From: James Gardner Date: Fri, 18 Aug 2023 17:01:50 +0100 Subject: [PATCH] Optimizers.scheduler_step() bug fix (#2352) optimizer bug fix --- nerfstudio/engine/optimizers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: