IBM / BAGAN

Keras implementation of Balancing GAN (BAGAN) applied to the MNIST example.
Eclipse Public License 1.0
107 stars 35 forks source link

Problem with initialisation of GAN. #2

Closed siddharthdivi closed 6 years ago

siddharthdivi commented 6 years ago

After the line, " print("BAGAN autoenc initialized, init gan") ", when the function self.init_gan() is called, an error occurs. The error occurs because it goes into the except block of the init_gan() function, the reason being, the function " self._get_lst_bck_name () " returns None.

Is this supposed to happen? As of now, I've just called the self.backup_point(0) to initialize the weights of the GAN. Is this alright ? This is just initialisation of the GAN right?

ova-mariani commented 6 years ago

It seems to have something to do with model saving and reloading.

If you train with the bagantrain.py script, a directory to store results including models should be automatically create `resdmodegmodeunbalanceepochslrseed`, is it?

Please send me more info about how you run the code to help me understanding what generates the issue.