keras 在train_on_batch中启用tensorboard