projects
/
pytorch.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[pytorch.git]
/
minidiffusion.py
diff --git
a/minidiffusion.py
b/minidiffusion.py
index
6855752
..
a386a12
100755
(executable)
--- a/
minidiffusion.py
+++ b/
minidiffusion.py
@@
-48,6
+48,7
@@
alpha_bar = alpha.log().cumsum(0).exp()
sigma = beta.sqrt()
for k in range(nb_epochs):
sigma = beta.sqrt()
for k in range(nb_epochs):
+
acc_loss = 0
optimizer = torch.optim.Adam(model.parameters(), lr = 1e-4 * (1 - k / nb_epochs) )
acc_loss = 0
optimizer = torch.optim.Adam(model.parameters(), lr = 1e-4 * (1 - k / nb_epochs) )