Change input from three-channel color to grayscale，use：self.turn = transforms.Grayscale(num_output_channels=1).
The same program can run through on one computer, and if it is changed to another computer (with the same environment), it will prompt an error:img should be PIL Image. Got <class ‘torch.Tensor’>。Who can help me answer