`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 |
|