LR removed from AdaCoor.

This commit is contained in:
alexds9 2023-11-17 20:52:15 +02:00
parent 5dc9f18061
commit fc6a4a7c27
2 changed files with 1 additions and 2 deletions

View File

@ -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=-group['lr'])
p.data.addcdiv_(gt_hat, denom, value=-1)
return loss

View File

@ -461,7 +461,6 @@ class EveryDreamOptimizer():
opt_class = AdaCoor
optimizer = opt_class(
itertools.chain(parameters),
lr=curr_lr,
eps=epsilon
)