checkpoint_path = '/tmp/datasets/model.hdf5'
checkpoint_dir = os.path.dirname(checkpoint_path)
#create a callback that saves the model
cp_callback = tf.keras.callbacks.ModelCheckpoint(filepath=checkpoint_path,
                                                 save_best_only=True,
                                                 save_weights_only=False,
                                                 verbose=0)