This repository was archived by the owner on Apr 17, 2023. It is now read-only.
This repository was archived by the owner on Apr 17, 2023. It is now read-only.
Bug with grayscale image normalization #122
Open
Description
While running PGAN to produce grayscale output, I think I found a bug in this line -- after the grayscale transform there's only one channel, so Transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
fails with RuntimeError: output with shape [1, 4, 4] doesn't match the broadcast shape [3, 4, 4]
. I fixed this by adding transformList[-1] = Transforms.Normalize([0.5], [0.5])
after line 423.
Metadata
Metadata
Assignees
Labels
No labels