这是我的自定义调用黑色：

class CustomCallback(keras.callbacks.Callback):
    def on_train_begin(self, logs=None):
          print("on_train_begin")
    def on_train_batch_end(self, batch, logs=None):
          print("on_train_batch_end")
    def on_batch_end(self, batch, logs=None):
          print("on_batch_end")

问题：在培训过程中，我只看到输出中出现"on_train_begin"。
我的Tensorflow版本：1.15我的Keras版本：2.2.5
有办法吗？
谢谢
注：我不想改变我的Keras/TF版本