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

def on_train_end(self, callback_data, model):

    callback_data['time/train/end_time'][0] = time.time()
    callback_data['time/train/end_time'].attrs['units'] = 'seconds'
