search

d-li14 / involution

[CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator
https://arxiv.org/abs/2103.06255
MIT License
1.31k stars 177 forks source link

GPU memory is large when the initialization？ #27

Closed pikaqiuEric closed 3 years ago

pikaqiuEric commented 3 years ago

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 commented 3 years ago

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.