Since you’ve mentioned you are using a CNN, I assume you are using nn.Conv3d layers and the input has a channel dimension of 4000?
Could you post your model architecture and post your GPU (including the memory)?