def METHOD_NAME(self, d_optimizer, g_optimizer, loss_fn, jit_compile=False):
    super(GAN, self).METHOD_NAME(jit_compile=jit_compile)
    self.d_optimizer = d_optimizer
    self.g_optimizer = g_optimizer
    self.loss_fn = loss_fn
    self.d_loss_metric = keras.metrics.Mean(name="d_loss")
    self.g_loss_metric = keras.metrics.Mean(name="g_loss")