I am using a pretrained model for extracting features(tf.keras) for images during the training phase and  running this in a GPU environment.
After the execution gets completed, i would like to release the GPU memory automatically without any manual intervention.