rosinality/style-based-gan-pytorch

Discriminator missing an activation function after the first convolutional layer

adambielski opened this issue · 2 comments

adambielski commented

After from_rgb layer in the discriminator, there is no activation function. A leaky relu nonlinearity should be applied there.

style-based-gan-pytorch/model.py

Line 552 in e5caa16

out = self.from_rgb[index](input)

rosinality commented

Thank you, I missed it. I will update the repo.

  • ❤️1
rosinality commented

Added this as an option at b160dbc for checkpoint compatibility. I will update checkpoint later.