Regarding the initialization of the class 'GeneratorWithMap'

[yunfan0621]

Hi,

Thank you for sharing the code of the great work!

I have been adopt the code of the generator to my own project, and I am concerned with the following line of code:

StyleRenderer/model.py

Line 216 in a1a093a

self.to_rgbs.append(ToRGB(out_channel, style_dim))

I feel this line is unnecessary as the to_rgb layers are already initialized in

StyleRenderer/model.py

Line 192 in a1a093a

super(GeneratorWithMap, self)._initialize_styled( \

If I understand it corretly, the current version of code will double the amount of to_rgb layers, but only the first half are used in the forward process (the generation results are still good though).

Best regards