`generate` is now named `decode`
Browse files
model.py
CHANGED
@@ -184,7 +184,7 @@ class VAE(tf.keras.Model):
|
|
184 |
z_sample = mu + sd * tf.random.normal(shape=(120,))
|
185 |
return z_sample, mu, sd
|
186 |
|
187 |
-
def
|
188 |
"""
|
189 |
Decode a latent representation of a song.
|
190 |
|
|
|
184 |
z_sample = mu + sd * tf.random.normal(shape=(120,))
|
185 |
return z_sample, mu, sd
|
186 |
|
187 |
+
def decode(self, z_sample: tf.Tensor=None) -> tf.Tensor:
|
188 |
"""
|
189 |
Decode a latent representation of a song.
|
190 |
|