Watch talks from the 2019 TensorFlow Dev Summit Watch now

tf.contrib.gan.losses.stargan_generator_loss_wrapper

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

Defined in tensorflow/contrib/gan/python/losses/python/tuple_losses_impl.py.

Convert a generator loss function to take a StarGANModel.

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

Args:

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

Returns:

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