data-efficient-gans
data-efficient-gans copied to clipboard
How to train StyleGAN2 with labeled dataset
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?
@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.