Hi, sorry for answering late. I notice that it invoke cuda.close(). I suppose that I could not reallocate GPU memory If I did that. Am I right? In fact, I only want to free some of the memory related to the model I want to delete, but not all. Moreover, my model does not involve tf.keras.Model; I use a custom training loop