Update model.py

model.py

@@ -112,9 +112,9 @@ class VAECost:
 
 class VAE(tf.keras.Model):
 
-  def __init__(self, dim_z, seed=2000, analytic_kl=True, name="autoencoder", **kwargs):
+  def __init__(self, dim_z=120, seed=2000, analytic_kl=True, name="autoencoder", **kwargs):
     super(VAE, self).__init__(name=name, **kwargs)
-    self.dim_x = (3, CAP, 1)
+    self.dim_x = (3, _CAP, 1)
     self.dim_z = dim_z
     self.seed = seed
     self.analytic_kl = analytic_kl