From 1b233e91ee4b6bccdf7ba17819a0c810f96ef7de Mon Sep 17 00:00:00 2001 From: lizz Date: Mon, 6 Nov 2017 00:04:32 +0800 Subject: [PATCH] Update dcgan.py --- models/dcgan.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/dcgan.py b/models/dcgan.py index 1dd8dbf..d8c7cc0 100644 --- a/models/dcgan.py +++ b/models/dcgan.py @@ -72,7 +72,7 @@ def __init__(self, isize, nz, nc, ngf, ngpu, n_extra_layers=0): main.add_module('initial.{0}.relu'.format(cngf), nn.ReLU(True)) - csize, cndf = 4, cngf + csize, cngf = 4, cngf while csize < isize//2: main.add_module('pyramid.{0}-{1}.convt'.format(cngf, cngf//2), nn.ConvTranspose2d(cngf, cngf//2, 4, 2, 1, bias=False)) @@ -169,7 +169,7 @@ def __init__(self, isize, nz, nc, ngf, ngpu, n_extra_layers=0): main.add_module('initial.{0}.relu'.format(cngf), nn.ReLU(True)) - csize, cndf = 4, cngf + csize, cngf = 4, cngf while csize < isize//2: main.add_module('pyramid.{0}-{1}.convt'.format(cngf, cngf//2), nn.ConvTranspose2d(cngf, cngf//2, 4, 2, 1, bias=False))