Run time error.output with shape [1, 256, 192] doesn't match the broadcast shape [3, 256, 192]

[carfei]

I also encountered the same problem when executing the program. I wonder if the author who had the same problem before has solved it? How to solve it? I really want to hear from you

[farfun]

transformer = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.5,0.5,0.5), (0.5,0.5,0.5)]) to transformer = transforms.Compose([transforms.ToTensor(), transforms.Normalize(0.5, 0.5)])