From 14775b026b59f04d8bd235cb55a68142c519639e Mon Sep 17 00:00:00 2001 From: "lukas.mosser" Date: Tue, 18 Jul 2017 09:43:28 +0100 Subject: [PATCH] Fixed ndf=ngf bug in main.py --- code/pytorch/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/code/pytorch/main.py b/code/pytorch/main.py index 2a9fb453..277224b0 100644 --- 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))