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):
+
     acc_loss = 0
     optimizer = torch.optim.Adam(model.parameters(), lr = 1e-4 * (1 - k / nb_epochs) )