tdeboissiere / DeepLearningImplementations

Implementation of recent Deep Learning papers
MIT License
1.81k stars 650 forks source link

Custom batch size for InfoGAN #20

Closed mridulbirla closed 7 years ago

mridulbirla commented 7 years ago

For issue #19 We need to pass batch size also when calling generator to predict an image. Following are the changes:

  1. In data_utils, Function get_disc_batch and plot_generated_batch call's predict function on generator model. As per Keras documentation(https://keras.io/models/sequential/) , we need to pass batch_size variable. Previously it didn't gave error because batch_size was 32 and predict function default batch_size is 32.
tdeboissiere commented 7 years ago

Good catch, thanks !