From f0b9e2086d556aea09b304fe99791e0348e1204e Mon Sep 17 00:00:00 2001 From: Martin Arjovsky Date: Mon, 30 Jan 2017 17:22:02 -0500 Subject: [PATCH] No BN only on gen --- main.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/main.py b/main.py index 16ff056..e4e5fce 100644 --- a/main.py +++ b/main.py @@ -116,10 +116,7 @@ if opt.netG != '': # load checkpoint if needed netG.load_state_dict(torch.load(opt.netG)) print(netG) -if opt.noBN: - netD = dcgan.DCGAN_D_nobn(opt.imageSize, nz, nc, ndf, ngpu, n_extra_layers) - netD.apply(weights_init) -elif opt.mlp_D: +if opt.mlp_D: netD = mlp.MLP_D(opt.imageSize, nz, nc, ndf, ngpu) else: netD = dcgan.DCGAN_D(opt.imageSize, nz, nc, ndf, ngpu, n_extra_layers)