From 9338cb78aa1b8260d050615f5473c5a23cae3108 Mon Sep 17 00:00:00 2001
From: Francois Fleuret <francois@fleuret.org>
Date: Fri, 12 Aug 2022 10:32:43 +0200
Subject: [PATCH] Update.
---
minidiffusion.py | 1 +
1 file changed, 1 insertion(+)
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) )
--
2.39.5