kazizzad / DCGAN-Gluon-MxNet

12 stars 2 forks source link

training on different dataset #2

Open Ashbajawed opened 5 years ago

Ashbajawed commented 5 years ago

can this code run on dataset other then celebA or cifer

kazizzad commented 5 years ago

Hey, sure it will, but you just need to load the data set you have in mind. Make sure you get the transform right as well.

Note. You need to also look for slightly different hyperparameters.