GPU memory is large when the initialization？

[pikaqiuEric]

I want to adding the module to the baskbone in the conv2.In operation, I found a phenomenon batchsize >4,the Memory overflow。

[d-li14]

Is it possible to specify your experimental details, such as the input image size, network type, GPU type you are using, and so on? Without this information, it is hard to diagnose.