From 9338cb78aa1b8260d050615f5473c5a23cae3108 Mon Sep 17 00:00:00 2001 From: Francois Fleuret 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.20.1