def create_discriminator(opt):
# Initialize the network
discriminator_a = network.PatchDiscriminator70(opt)
discriminator_b = network.PatchDiscriminator70(opt)
# Init the network
network.weights_init(discriminator_a, init_type = opt.init_type, init_gain = opt.init_gain)
network.weights_init(discriminator_b, init_type = opt.init_type, init_gain = opt.init_gain)
print('Discriminators is created!')
return discriminator_a, discriminator_b
def create_discriminator(opt):
# Initialize the networks
discriminator = network.PatchDiscriminator70(opt)
# Init the networks
network.weights_init(discriminator,
init_type=opt.init_type,
init_gain=opt.init_gain)
return discriminator
