From 692ddefe8186de2493d3b20bc02204d4d5700000 Mon Sep 17 00:00:00 2001 From: Lalit Bhagat <36616618+lalitbhagat7@users.noreply.github.com> Date: Mon, 26 Jun 2023 18:09:48 -0700 Subject: [PATCH] add retain_graph=True in Discriminator --- ML/Pytorch/GANs/2. DCGAN/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ML/Pytorch/GANs/2. DCGAN/train.py b/ML/Pytorch/GANs/2. DCGAN/train.py index aa943682..3980aa79 100644 --- a/ML/Pytorch/GANs/2. DCGAN/train.py +++ b/ML/Pytorch/GANs/2. DCGAN/train.py @@ -77,7 +77,7 @@ loss_disc_fake = criterion(disc_fake, torch.zeros_like(disc_fake)) loss_disc = (loss_disc_real + loss_disc_fake) / 2 disc.zero_grad() - loss_disc.backward() + loss_disc.backward(retain_graph=True) opt_disc.step() ### Train Generator: min log(1 - D(G(z))) <-> max log(D(G(z))