change comment

mori360 · Jan 30, 2025 · 12a0bb2 · 12a0bb2
1 parent 4d776f0
commit 12a0bb2
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/torchtitan/optimizer.py b/torchtitan/optimizer.py
@@ -180,8 +180,7 @@ def step(self) -> None:
             scheduler.step()
 
     def state_dict(self) -> Dict[str, Any]:
-        # Right now we have lr_scheduler with the same state_dict for all optimizers,
-        # so we can just save one.
+        # We have lr_scheduler with the same state_dict for all optimizers, so can just save one.
         assert (
             len(self.schedulers) > 0
         ), "Must have at least one scheduler to save state_dict"