2 bugs correction in VAE :-)

VAE/modules/VAE.py

@@ -43,7 +43,7 @@ class VAE(keras.Model):
 
         
     def call(self, inputs):
-        z = self.encoder(inputs)
+        z_mean, z_log_var, z = self.encoder(inputs)
         y_pred = self.decoder(z)
         return y_pred
                 

VAE/modules/callbacks.py

@@ -31,6 +31,8 @@ class ImagesCallback(Callback):
         
     def on_epoch_end(self, epoch, logs={}):  
         
+        if epoch<1 : return
+        
         # ---- Get latent points
         #
         if self.x is None:
@@ -72,6 +74,7 @@ class BestModelCallback(Callback):
         self.loss = np.Inf
         
     def on_epoch_end(self, epoch, logs=None):
+        if epoch<1 : return
         current = logs.get("loss")
         if current<self.loss:
             self.loss = current

fidle/log/finished.json

@@ -78,8 +78,8 @@
         "duration": "00:00:03 329ms"
     },
     "VAE1": {
-        "path": "/home/pjluc/dev/fidle/VAE",
-        "start": "Monday 4 January 2021, 17:59:08",
+        "path": "/gpfsdswork/projects/rech/mlh/uja62cb/fidle/VAE",
+        "start": "Tuesday 5 January 2021, 00:07:12",
         "end": "",
         "duration": "Unfinished..."
     },
@@ -150,8 +150,8 @@
         "duration": "00:00:08 736ms"
     },
     "VAE8": {
-        "path": "/home/pjluc/dev/fidle/VAE",
-        "start": "Monday 4 January 2021, 23:53:16",
+        "path": "/gpfsdswork/projects/rech/mlh/uja62cb/fidle/VAE",
+        "start": "Tuesday 5 January 2021, 00:23:42",
         "end": "",
         "duration": "Unfinished..."
     }