I want to remove all instance of torch from my gpu to train the same model with different hyperparemter in google colab #11279
Unanswered
Rami-Ismael
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
I was on Google Colab and finding that I could train my model one times, but that on the 2th time I’d run into the memory error. Using torch.cuda.empty_cache() between runs did not help. All I could do was restart my kernel.
Beta Was this translation helpful? Give feedback.
All reactions