How to train StyleGAN2 with labeled dataset

[TccccD]

Hello, I found that compared with the native stylegan2, your code specifically removes the label. I want to ask you whether you have tried to add the label for training, how is the effect?

[TccccD]

@zsyzzsoft

[zsyzzsoft]

I did try conditional generation before, and you will be able to do it if you make some changes in loss.py (e.g. something like real_scores = reduce_sum(D(reals) * one_hot(label)). Or you may try the PyTorch version to run conditional generation.