keras API change

[whimian]

keras callbacks has a new API. on_train_batch_begin and on_train_batch_end are the new names for on_batch_begin and on_batch_end.

Refer to:https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/keras/callbacks.py#L175