tensorflow-mnist-VAE icon indicating copy to clipboard operation
tensorflow-mnist-VAE copied to clipboard

Training on all the digits at once

Open udion opened this issue 6 years ago • 3 comments

Hi, While training on multiple images of MNIST I was not getting any reasonable output upon generation of images, however when I trained on a single digit (say 3) I was able to generate reasonable looking digits (like 3).

I tried increasing the iteration for training with multiple digits in dataset together, but still I am not getting reasonable outputs like yours, any suggestions?

udion avatar Jan 09 '18 07:01 udion

Hi, from my experience, I suggest you to use deeper architectures. Generated images are a bit sensitive to network architecture.

DCGAN like architecture would be good start point.

hwalsuklee avatar Jan 11 '18 13:01 hwalsuklee

I have tried dcgan as well, but it also doesn't when dataset has mixture of all the digits. any suggestions?

udion avatar Jan 14 '18 20:01 udion

@udion can you let me know how can I train network just one class and generate one class output?

lerndeep avatar May 14 '20 02:05 lerndeep