Hmm. I want to avoid restarting the kernel. I’d like to train all the n-models in one go. Is that possible? torch.cuda.empty_cache() seems to work. I basically call it after training each model. But I’m not sure what it’s clearing.