TensorFlow 2.0中自定义训练循环中使用的自定义回叫

Question

我正在尝试为使用TensorFlow 2.0实现的自定义训练循环实现自定义回调。可以访问：https://www.tensorflow.org/guide/keras/custom_callback。假定给出的代码与不同的tf.keras.Model方法相关。关于如何将其纳入自定义培训循环的任何指导？谢谢与问候。