Unequal Dataset Dimension

[mudabir94]

I have a spectral cube dataset of shape (100,256,256,50).
There are 100 images of 256 by 256 of 50 dimensions. I want to use such kind of data on the 3D network but according to my understanding the size of height, width and dimensions should be equal otherwise the model won't compile. Please guide what to do in this scenario. Thank you.

[ZFTurbo]

First 3 dimensions must be divisible on 32. So something like (96,256,256,50) will be ok.
Also have in mind that it will require a lot of GPU memory. If you have 48 GB then it probably fit, otherwise I'm not sure.