ValueError: Expected tensor to be a tensor image of size (C, H, W). Got tensor.size() = torch.Size([8, 8])

I answered your question abt normalization in the other thread to maintain continuity.