def build_img_discriminator(args, vocab): discriminator = None d_kwargs = {} d_weight = args.discriminator_loss_weight d_img_weight = args.d_img_weight if d_weight == 0 or d_img_weight == 0: return discriminator, d_kwargs d_kwargs = { 'arch': args.d_img_arch, 'normalization': args.d_normalization, 'activation': args.d_activation, 'padding': args.d_padding } if args.sg_context_dim_d > 0: #discriminator has access to pooled GCNN features layout_dim = 0 ########################### #discriminator has access to layout if args.layout_for_discrim: layout_dim = args.gconv_dim if args.image_patch_discr: discriminator = CondGANPatchDiscriminator( **d_kwargs, layout_dim=layout_dim, sg_context_dim=args.sg_context_dim_d) else: discriminator = CondGANDiscriminator( **d_kwargs, layout_dim=layout_dim, sg_context_dim=args.sg_context_dim_d) else: discriminator = PatchDiscriminator(**d_kwargs) return discriminator, d_kwargs
def build_img_discriminator(args, vocab): discriminator = None d_kwargs = {} d_weight = args.discriminator_loss_weight d_img_weight = args.d_img_weight if d_weight == 0 or d_img_weight == 0: return discriminator, d_kwargs d_kwargs = { 'arch': args.d_img_arch, 'normalization': args.d_normalization, 'activation': args.d_activation, 'padding': args.d_padding, } discriminator = PatchDiscriminator(**d_kwargs) return discriminator, d_kwargs