X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=minidiffusion.py;h=a386a12c8b34b3fc848b0afc93a94089b33af9a6;hb=9338cb78aa1b8260d050615f5473c5a23cae3108;hp=6855752e6e67b2e2c53b179617685caf715f24b8;hpb=317cc211cf9589a9eee5d937f0d0182719f24790;p=pytorch.git

diff --git a/minidiffusion.py b/minidiffusion.py
index 6855752..a386a12 100755
--- 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):
+
     acc_loss = 0
     optimizer = torch.optim.Adam(model.parameters(), lr = 1e-4 * (1 - k / nb_epochs) )