From 26401258fef8d9534fcb5d46f8e5d06cf5bf1b68 Mon Sep 17 00:00:00 2001 From: Anthony Tatum Date: Wed, 22 Mar 2023 13:49:13 -0400 Subject: [PATCH 1/2] Update training.py --- SinGAN/training.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/SinGAN/training.py b/SinGAN/training.py index 645a8635..e7d89d65 100755 --- a/SinGAN/training.py +++ b/SinGAN/training.py @@ -107,8 +107,9 @@ def train_single_scale(netD,netG,reals,Gs,Zs,in_s,NoiseAmp,opt,centers=None): ########################### for j in range(opt.Dsteps): # train with real - netD.zero_grad() - + # netD.zero_grad() + for param in netD.parameters(): + param.grad = None output = netD(real).to(opt.device) #D_real_map = output.detach() errD_real = -output.mean()#-a @@ -171,7 +172,9 @@ def train_single_scale(netD,netG,reals,Gs,Zs,in_s,NoiseAmp,opt,centers=None): ########################### for j in range(opt.Gsteps): - netG.zero_grad() + # netG.zero_grad() + for param in netG.parameters(): + param.grad = None output = netD(fake) #D_fake_map = output.detach() errG = -output.mean() From 89e5a1a1da9102c7cc46f84bd9363e4a7444df03 Mon Sep 17 00:00:00 2001 From: Anthony Tatum Date: Wed, 22 Mar 2023 13:50:06 -0400 Subject: [PATCH 2/2] Update models.py --- SinGAN/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/SinGAN/models.py b/SinGAN/models.py index 99a8dfe3..6ba35d0c 100755 --- a/SinGAN/models.py +++ b/SinGAN/models.py @@ -7,7 +7,7 @@ class ConvBlock(nn.Sequential): def __init__(self, in_channel, out_channel, ker_size, padd, stride): super(ConvBlock,self).__init__() - self.add_module('conv',nn.Conv2d(in_channel ,out_channel,kernel_size=ker_size,stride=stride,padding=padd)), + self.add_module('conv',nn.Conv2d(in_channel ,out_channel,kernel_size=ker_size,stride=stride,padding=padd, bias=False)), self.add_module('norm',nn.BatchNorm2d(out_channel)), self.add_module('LeakyRelu',nn.LeakyReLU(0.2, inplace=True))