Fixed ndf=ngf bug in main.py

LukasMosser · Jul 18, 2017 · 14775b0 · 14775b0
1 parent 0d4839a
commit 14775b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/code/pytorch/main.py b/code/pytorch/main.py
@@ -88,7 +88,7 @@ def weights_init(m):
     netG.load_state_dict(torch.load(opt.netG))
 print(netG)
 
-netD = dcgan.DCGAN3D_D(opt.imageSize, nz, nc, ngf, ngpu)
+netD = dcgan.DCGAN3D_D(opt.imageSize, nz, nc, ndf, ngpu)
 netD.apply(weights_init)
 if opt.netD != '':
     netD.load_state_dict(torch.load(opt.netD))