@alex.veuthey, @mmisiur, @Shisho_Sama
Maybe I gave you wrong information.
I just forgot to mention that my data is 4D.
For 32 batch size, train data shape:
[32, 6, 512, 512]
Actually, I am using Conv2d for this 4D data. Maybe this is wrong.
Can you please tell me how can I fix it?
I have no idea how to handle 4D data for CNN?
Please Help