Saved model size 57MB but GPU usage 13GB

I'm doing a deep learning run on Google collab. Its a UNet. The saved model's size is 57MB ( saved with \`[torch.save](https://torch.save)(model.state\_dict(), "[model.pt](https://model.pt)" )\` but the GPU shoots up to 13GB. The batch size of data is small, with less than a megabyte of data and target being trained at once.

Does a Unet in pytorch really consume that much GPU during training?

Is there a decent approximation to a model's "training size" that I could be doing ahead of …

approximation deeplearning gpu pytorch training unet