From edf0f8582edd3bae0273df03f6a219aa5e210042 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 2 Jun 2020 11:15:55 -0600 Subject: [PATCH] Fix rrdb bug --- codes/models/networks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/networks.py b/codes/models/networks.py index 137a3361..bce2e531 100644 --- a/codes/models/networks.py +++ b/codes/models/networks.py @@ -27,7 +27,7 @@ def define_G(opt, net_key='network_G'): # Need to adjust the scale the generator sees by the stride since the stride causes a down-sample. gen_scale = scale * initial_stride netG = RRDBNet_arch.RRDBNet(in_nc=opt_net['in_nc'], out_nc=opt_net['out_nc'], - nf=opt_net['nf'], nb=opt_net['nb'], scale=scale, initial_stride=initial_stride) + nf=opt_net['nf'], nb=opt_net['nb'], scale=gen_scale, initial_stride=initial_stride) elif which_model == 'AssistedRRDBNet': netG = RRDBNet_arch.AssistedRRDBNet(in_nc=opt_net['in_nc'], out_nc=opt_net['out_nc'], nf=opt_net['nf'], nb=opt_net['nb'], scale=scale)