Using -1.0 instead of -1.

2023-11-17 21:55:12 +02:00 · 2023-11-17 21:55:12 +02:00 · 32ee7cb7f0
parent dcf2969640
commit 32ee7cb7f0
1 changed files with 2 additions and 2 deletions
--- a/optimizer/adacoor.py
+++ b/optimizer/adacoor.py
@ -32,6 +32,6 @@ class AdaCoor(torch.optim.Optimizer):
                    gt_hat = (epsilon * p.grad.data).to(dtype=torch.float32, device=p.device)

                    denom = vt.sqrt().add_(group['epsilon']).to(dtype=p.dtype, device=p.device)
-                    p.data.addcdiv_(gt_hat, denom, value=-1)
-                
+                    p.data.addcdiv_(gt_hat, denom, value=-1.0)
+
        return loss