Train on unconditional dataset?

brain-research / self-attention-gan

Apache License 2.0

998 stars 175 forks source link

Train on unconditional dataset? #15

Open suenlaries opened 5 years ago

suenlaries commented 5 years ago

How can we train on Unconditional dataset? I have tried on CelebA dataset, with set parameter number_class to 1. But training process not going well, got error after 60K steps. How can I train with unconditional dataset?

capture1

capture

capture3