diff options
| -rw-r--r-- | cli/app/search/search_class.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py index 139ca36..0fcfdfd 100644 --- a/cli/app/search/search_class.py +++ b/cli/app/search/search_class.py @@ -88,7 +88,7 @@ def find_nearest_vector(sess, generator, opt_fp_in, opt_dims, out_images, out_la 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.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_y = tf.compat.v1.Variable(y_initial, dtype=np.float32, constraint=lambda t: tf.clip_by_value(t, 0, 1.5)) input_trunc = tf.compat.v1.constant(1.0) output = generator({ 'z': input_z, |