Shouldn't "self.num_ws += block.num_torgb" happen for every Block? To my understanding, every TORGB layer (not just the one in the last block) should receive a ws code, and thus, "num_ws" should be "\sum_i (block_i.num_conv+block_i.num_torgb)". I may miss something here. It would be great if this can be clarified. Thank!
Hi, Thanks for the great contributions.
For the StyleGAN2 backbone, I find this part hard to understand: https://github.com/NVlabs/eg3d/blob/493076071bc400be3289dd6725ef1eef0a552a9d/eg3d/training/networks_stylegan2.py#L499
Shouldn't "self.num_ws += block.num_torgb" happen for every Block? To my understanding, every TORGB layer (not just the one in the last block) should receive a ws code, and thus, "num_ws" should be "\sum_i (block_i.num_conv+block_i.num_torgb)". I may miss something here. It would be great if this can be clarified. Thank!