def METHOD_NAME(self):
    super().METHOD_NAME()
    # Make sure all the models created in keras has been deleted and cleared
    # from the global keras grpah, also do a force GC to recycle the GPU
    # memory.
    tf.keras.backend.clear_session()
    gc.collect()