openai / improved-gan

Code for the paper "Improved Techniques for Training GANs"

Home Page:https://arxiv.org/abs/1606.03498

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

how to specify num of classes or labels in imagenet code

SHANKARMB opened this issue · comments

I have a custom dataset like cifar10 dataset... How do I specify the num of classes or labels??
Though we can pass it to y_dim, but it is not being used anywhere..