I am using google collab(GPU) to run my model but as I create an object of my model, collab crash occurs. After analysing I find that it breaks down whenever I try to increase the size of the FC layer(Linear).
I am not getting why it crashes only on it.
My model contains layers such as:
1.Conv2d
2.maxpool
3.Conv3d
4.Linear layers
5.activations(leakyReLu)
But it only crashes due to change in Linear layer output parameters. I am not getting how all the memory could be taken by Linear layers (weights and bias).