使用train_on_batch进行Keras回调？

Question

使用train_on_batch时有没有办法使用keras回调？

我不喜欢从头开始编写适合循环。关于如何使用回调机制的任何想法？