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