How to keep track of GPU memory on google colab?

Question

I am trying to train a deep neural network (DNN) on Google Colab with the use of the PyTorch framework.

So far, I am debugging my network, a