TensorFlow 2.0 RC is available Learn more

tf.contrib.gan.losses.stargan_discriminator_loss_wrapper

View source on GitHub

Convert a discriminator loss function to take a StarGANModel.

tf.contrib.gan.losses.stargan_discriminator_loss_wrapper(loss_fn)

The new function has the same name as the original one.

Args:

  • loss_fn: A python function taking Discriminator's real/fake prediction for real data and generated data.

Returns:

A new function that takes a StarGANModel namedtuple and returns the same loss.