diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-08 17:54:36 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-08 17:54:36 +0100 |
| commit | 8fc7f19a632408a59134ecc092d21d6a31fbb723 (patch) | |
| tree | 6f06d10c99f2a083f07f34d306a584b4256c5989 /cli/app/search/search_class.py | |
| parent | 034922d32c1d9df996e6292a17fb4fb4cb04395d (diff) | |
more latent tweaking
Diffstat (limited to 'cli/app/search/search_class.py')
| -rw-r--r-- | cli/app/search/search_class.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py index 3a56f68..a41a141 100644 --- a/cli/app/search/search_class.py +++ b/cli/app/search/search_class.py @@ -87,8 +87,8 @@ def find_nearest_vector(sess, generator, opt_fp_in, opt_dims, out_images, out_la z_initial = truncated_z_sample(batch_size, z_dim, truncation/2) y_initial = create_labels_uniform(batch_size, vocab_size) - input_z = tf.compat.v1.Variable(z_initial, dtype=np.float32, constraint=lambda t: tf.clip_by_value(t, -2, 2)) - input_y = tf.compat.v1.Variable(y_initial, dtype=np.float32, constraint=lambda t: tf.clip_by_value(t, 0, 1)) + input_z = tf.compat.v1.Variable(z_initial, dtype=np.float32, constraint=lambda t: tf.clip_by_value(t, -2.5, 2.5)) + input_y = tf.compat.v1.Variable(y_initial, dtype=np.float32, constraint=lambda t: tf.clip_by_value(t, -0.5, 1.5)) input_trunc = tf.compat.v1.constant(1.0) output = generator({ 'z': input_z, |