From 55f2764fef432226d89a858443154ae296e3adab Mon Sep 17 00:00:00 2001 From: James Betker Date: Thu, 1 Oct 2020 11:28:18 -0600 Subject: [PATCH] Allow fixup50 to be used as a discriminator --- codes/models/networks.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/codes/models/networks.py b/codes/models/networks.py index 5f28cf72..22dea7d7 100644 --- a/codes/models/networks.py +++ b/codes/models/networks.py @@ -120,6 +120,8 @@ def define_D_net(opt_net, img_sz=None, wrap=False): netD = GradDiscWrapper(netD) elif which_model == 'discriminator_resnet': netD = DiscriminatorResnet_arch.fixup_resnet34(num_filters=opt_net['nf'], num_classes=1, input_img_size=img_sz) + elif which_model == 'discriminator_resnet_50': + netD = DiscriminatorResnet_arch.fixup_resnet50(num_filters=opt_net['nf'], num_classes=1, input_img_size=img_sz) elif which_model == 'discriminator_resnet_passthrough': netD = DiscriminatorResnet_arch_passthrough.fixup_resnet34(num_filters=opt_net['nf'], num_classes=1, input_img_size=img_sz, number_skips=opt_net['number_skips'], use_bn=True,