Should the grayscale transform come last in the list?

[diviramon]

pytorch_GAN_zoo/models/trainer/gan_trainer.py

Line 423 in b75dee4

transformList = [Transforms.Grayscale(1)] + transformList

As it is right now, it will through an error when using 1 channel images since they are converted to GrayScale first then a transform on three channels (0.5,0.5,0.5) is applied.

[GarySarwin]

Try this:

transformList = [Transforms.Grayscale(1)] + transformList[:len(transformList)-1] + [Transforms.Normalize((0.5,),(0.5,))]

[aaroncoyner]

This fixed it for me: #122 (comment)