-
Haitang Hu authored
This would match the behavior described in on_batch_end() functions in Keras callback. See https://www.tensorflow.org/api_docs/python/tf/keras/callbacks/Callback. PiperOrigin-RevId: 293887583
67f6015a
This would match the behavior described in on_batch_end() functions in Keras callback. See https://www.tensorflow.org/api_docs/python/tf/keras/callbacks/Callback. PiperOrigin-RevId: 293887583