Open callmeYe opened 3 years ago
I ran this program on Colab, and when I changed the parameters and tried to run it again, I got an error with CUDA out of memory. I can only kill the python3 process and rerun the entire notebook.
torch.cuda.empty_cache()
I ran this program on Colab, and when I changed the parameters and tried to run it again, I got an error with CUDA out of memory. I can only kill the python3 process and rerun the entire notebook.