Gluon Out of Memory Issue (cudaMalloc failed: out of memory)

Hi, this sounds like you are (maybe) not reducing the size of the image with “pooling” operations? Can you please post the full architecture? In your code I cannot see the size of the convolution layers. This topic discusses a similar problem.