'Called when training is about to begin
Arguments:
callback_data (HDF5 dataset): shared data between callbacks
model (Model): model object'

def on_train_begin(self, callback_data, model, epochs):

    pass
