in models_all.py [0:0]
def forward(self, dlatents_in):
# Input: Disentangled latents (W) [minibatch, num_layers, dlatent_size].
for i, m in enumerate(self.blocks.values()):
if i == 0:
x = m(dlatents_in[:, 2 * i : 2 * i + 2])
else:
# continue
x = m(x, dlatents_in[:, 2 * i : 2 * i + 2])
rgb = self.torgb(x)
return rgb