update: reset

kozistr · May 5, 2024 · ae7dbed · ae7dbed
1 parent 06026db
commit ae7dbed
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/pytorch_optimizer/optimizer/sam.py b/pytorch_optimizer/optimizer/sam.py
@@ -594,9 +594,6 @@ def __str__(self) -> str:
     def reset(self):
         for group in self.param_groups:
             for p in group['params']:
-                if p.grad is None:
-                    continue
-
                 state = self.state[p]
 
                 state['s'] = torch.ones_like(p)