I was able to solve that error. But now its showing
RuntimeError: Given groups=1, weight of size [64, 3, 7, 7], expected input[5, 64, 70, 70] to have 3 channels, but got 64 channels instead.
I get the above mentioned error. I assume this is because input channels for RGB are 3.
The size of my patches are torch.Size([112, 64, 64, 64]).
How can I modify it for my own dataset. My images are dicom images.