update: code

kozistr · Aug 13, 2024 · 3f38e3f · 3f38e3f
1 parent 537a0d7
commit 3f38e3f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pytorch_optimizer/optimizer/prodigy.py b/pytorch_optimizer/optimizer/prodigy.py
@@ -111,8 +111,7 @@ def step(self, closure: CLOSURE = None) -> LOSS:
         if 'd_numerator' not in group:
             group['d_numerator'] = torch.tensor([0.0], device=device)
         elif group['d_numerator'].device != device:
-            # pragma: no cover
-            group['d_numerator'] = group['d_numerator'].to(device)
+            group['d_numerator'] = group['d_numerator'].to(device)  # pragma: no cover
 
         d_numerator = group['d_numerator']
         d_numerator.mul_(beta3)