I’ve been training VGG16 from scratch on the flower recognition dataset and after the first epoch itself CUDA runs out of memory. Is there a way out of this? I recently studies tensorflow.js and it had a command called tf.tidy() which would delete tensors as soon as their usage was over for proper memory usage. Does torch  have a provision like that?