Add that step starting at 1 is imposed by pytorch

VectorInstitute · Oct 31, 2024 · 5a6ac7a · 5a6ac7a
1 parent 3361ef7
commit 5a6ac7a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fl4health/utils/nnunet_utils.py b/fl4health/utils/nnunet_utils.py
@@ -462,7 +462,7 @@ def get_lr(self) -> Sequence[float]:
                 f"Current LR step of {self._step_count} reached Max Steps of {self.max_steps}. LR will remain fixed.",
             )
 
-        # Subtract 1 from step count since it starts at 1
+        # Subtract 1 from step count since it starts at 1 (imposed by PyTorch)
         curr_step = min(self._step_count - 1, self.max_steps)
         curr_window = int(curr_step / self.steps_per_lr)