Fixed Test error when encoder(image).

yunjey / pytorch-tutorial

PyTorch Tutorial for Deep Learning Researchers

MIT License

29.79k stars 8.03k forks source link

Fixed Test error when encoder(image). #166

Closed mdhasanai closed 4 years ago

mdhasanai commented 5 years ago

When we load an image for the sample, we have to make sure that the image has three color channel (RGB) because it might be grayscale. So we should convert it for sampling.