BUGZ

Smith42 · Jan 18, 2022 · a9da3d6 · a9da3d6
1 parent 68e6310
commit a9da3d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/denoising_diffusion_pytorch.py b/denoising_diffusion_pytorch.py
@@ -601,7 +601,7 @@ def train(self):
                 t1 = time()
                 with open(str(self.logdir / 'loss.txt'), 'a') as df:
                     df.write(f'{self.step},{loss.item()}\n')
-                loss.backward(loss / self.gradient_accumulate_every, self.opt)
+                (loss / self.gradient_accumulate_every).backward()
 
             self.opt.step()
             self.opt.zero_grad()