Retrieved name parameter

model.py

@@ -114,7 +114,7 @@ class VAECost:
 
 class VAE(tf.keras.Model):
 
-  def __init__(self, **kwargs):
+  def __init__(self, name="variational autoencoder", **kwargs):
     super(VAE, self).__init__(name=name, **kwargs)
     self.dim_x = (3, _CAP, 1)
     self.encoder = Encoder_Z(dim_z=120).build()