VAE : I'll get him anyway!

028b4797 · Jean-Luc Parouty Jean-Luc.Parouty@simap.grenoble-inp.fr · 9ede758b · 028b4797 · 028b4797 · 028b4797
Commit 028b4797 authored 5 years ago by Jean-Luc Parouty Jean-Luc.Parouty@simap.grenoble-inp.fr
--- a/VAE/03-Prepare-CelebA.ipynb
+++ b/VAE/03-Prepare-CelebA.ipynb
--- a/VAE/04-Check-CelebA.ipynb
+++ b/VAE/04-Check-CelebA.ipynb
--- a/VAE/05-VAE-with-CelebA.ipynb
+++ b/VAE/05-VAE-with-CelebA.ipynb
--- a/VAE/modules/data_generator.py
+++ b/VAE/modules/data_generator.py
@@ -106,7 +106,7 @@ class DataGenerator(Sequence):
    
    
    def on_epoch_end(self):
-        self.cluster_i = clusters_size
+        self.cluster_i = self.clusters_size
        self.read_next_cluster()
    
    

--- a/VAE/modules/vae.py
+++ b/VAE/modules/vae.py
@@ -234,8 +234,8 @@ class VariationalAutoencoder():
                                          shuffle = True,
                                          epochs = epochs,
                                          initial_epoch = initial_epoch,
-                                          callbacks = callbacks_list
-#                                           validation_data = (x_test, x_test)
+                                          callbacks = [], #callbacks_list,
+                                          validation_data = (x_test, x_test)
                                         )
        
        end_time  = time.time()