Keras documentation: WGAN-GP overriding `Model.train_step`
Keras documentation
keras.io