From: Francois Fleuret <francois@fleuret.org>
Date: Fri, 12 Aug 2022 08:32:43 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=9338cb78aa1b8260d050615f5473c5a23cae3108;p=pytorch.git

Update.
---

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) )