Added missing assignation
Browse files
model.py
CHANGED
|
@@ -177,7 +177,7 @@ class VAE(tf.keras.Model):
|
|
| 177 |
x_input = tf.expand_dims(x_input, axis=-1) # Add channel dimension
|
| 178 |
|
| 179 |
if tf.rank(x_input) == 3: # If there's no batch dimension, add it
|
| 180 |
-
tf.expand_dims(x_input, axis=0)
|
| 181 |
|
| 182 |
mu, rho = tf.split(self.encoder(x_input), num_or_size_splits=2, axis=1)
|
| 183 |
sd = tf.math.log(1 + tf.math.exp(rho))
|
|
|
|
| 177 |
x_input = tf.expand_dims(x_input, axis=-1) # Add channel dimension
|
| 178 |
|
| 179 |
if tf.rank(x_input) == 3: # If there's no batch dimension, add it
|
| 180 |
+
x_input = tf.expand_dims(x_input, axis=0)
|
| 181 |
|
| 182 |
mu, rho = tf.split(self.encoder(x_input), num_or_size_splits=2, axis=1)
|
| 183 |
sd = tf.math.log(1 + tf.math.exp(rho))
|