Oh okay - thanks very much! What changes would you recommend to self.fc / forward(self.inp)
such that I would get style’s size as 64x1?
real_out and the assoc. real-vs-fake ‘normal’ GAN loss work fine, so I don’t think the problem lies with self.conv or self.final_conv etc. Just need to get this additional classification loss working on top of the usual GAN loss.