semi-supervised-GAN icon indicating copy to clipboard operation
semi-supervised-GAN copied to clipboard

More than 200 labeled images are used to train

Open yongpi-scu opened this issue 6 years ago • 1 comments

Hi, Thanks for your work. I think there is something wrong in your result.In general, we want to use only 200 labeled images in training process.But in your code, you adopt the mnist dataloader provide by tensorflow. https://github.com/LDOUBLEV/semi-supervised-GAN/blob/52ba4a5ebb5ff07e6eceb48897ca78cb6b14f529/train.py#L120 In mnist.train.next_batch() function, there is a default param shuffle=True,so each epoch used different 200 labeled images.Consider overlapping, the number of labeled images is in [200, 200*epochs]. Am i understanding right? Looking forward to your reply. Thank you.

yongpi-scu avatar Oct 17 '18 04:10 yongpi-scu

@PYsaber Hi, thanks for your question. I set a flag to choose whether the label information works in the loss function of the network.You can check the code include 70thline and 119th line in the train.py. The label information are not used in the loss function when self.flag is equal 0.

LDOUBLEV avatar Nov 05 '18 15:11 LDOUBLEV