RuntimeError: CUDA out of memory. in colab code example

Example: RuntimeError: CUDA out of memory.

Your GPU is out of memory, 
reduce your batch size until your code runs without this error

# also, find any processes using gpu and kill them with: 
nvidia-smi
kill -9 [pid]