From d32a222b685ea5b4f340bc6706dbb761fe540cb4 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 12 Jan 2020 14:40:33 +0100 Subject: fix disentangled --- cli/app/search/live.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/cli/app/search/live.py b/cli/app/search/live.py index 9fcf91b..f066faf 100644 --- a/cli/app/search/live.py +++ b/cli/app/search/live.py @@ -207,9 +207,9 @@ class Interpolator: z_sum = lerp_z.output + sin_z.output # Latent - saturation - abs_zoom = InterpolatorParam(name='abs_zoom', value=1.0) - z_abs = z_sum / tf.abs(z_sum) * abs_zoom.variable - z_mix = LerpParam('abs_mix', a_in=z_sum, b_in=z_abs, shape=[BATCH_SIZE, Z_DIM], datatype="input") + saturation = InterpolatorParam(name='saturation', value=1.0) + z_abs = z_sum / tf.abs(z_sum) * saturation.variable + z_mix = LerpParam('saturation_mix', a_in=z_sum, b_in=z_abs, shape=[BATCH_SIZE, Z_DIM], datatype="input") # Latent - disentangled vectors zoom = InterpolatorParam(name='zoom', value=0.0).variable * self.disentangled['zoom'] @@ -278,7 +278,6 @@ class Interpolator: self.disentangled['b'] = disentangled_color[:, :, 2] self.disentangled['luminance'] = np.sum(disentangled_color, axis=2) - def get_feed_dict(self): opt = {} for param in self.opts.values(): -- cgit v1.2.3-70-g09d2