Thank you for your suggestion. However, I don’t have the biggest model because I am just trying to find out which model configuration will yield the highest accuracy for me and modify it. Is there a way to release the memory from the old memory block since I will train the modified model from scratch anyway?