davrot/pytorch-sbs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add files via upload

Browse source
This commit is contained in:
David Rotermund 2023-01-05 13:24:47 +01:00 committed by GitHub
parent 81ec320f5c
commit f12a1b1e75
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
network/Adam.py
View file

@ -60,6 +60,8 @@ class Adam(torch.optim.Optimizer):
state_steps = []
sbs_setting = []
assert len(self.param_groups) == 1
for id, p in enumerate(self.params):
if p.grad is not None:
params_with_grad.append(p)
@ -95,7 +97,7 @@ class Adam(torch.optim.Optimizer):
state_steps,
beta1=self.beta1,
beta2=self.beta2,
lr=self.lr,
lr=self.param_groups[0]["lr"],
eps=self.eps,
maximize=self.maximize,
)
@ -149,6 +151,6 @@ class Adam(torch.optim.Optimizer):
else:
delta = torch.exp(-step_size * (exp_avg / denom))
print(
f"{float(delta.min()) - 1.0:.4e} {float(delta.max()) - 1.0:.4e}"
f"{float(delta.min()) - 1.0:.4e} {float(delta.max()) - 1.0:.4e} {lr:.4e}"
)
param *= delta