do you mean something like?
        #RuntimeError: Given groups=1, weight of size [64, 3, 7, 7], expected input[64, 600, 800, 3] to have 3 channels, but got 600 channels instead
    
        images = images.permute(3,1)

I understand the following example but not sure how to apply to my situation above:
torch_permute1262×564 35 KB