summaryrefslogtreecommitdiff
path: root/cli/app/search/search_class.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-01-08 20:17:30 +0100
committerJules Laplace <julescarbon@gmail.com>2020-01-08 20:17:30 +0100
commitb9f515173f480a0468fc54d05bf134ebb15b0328 (patch)
treed473b3946ba20f95b31305e4c9c04ca17dba92df /cli/app/search/search_class.py
parent76d4a9ed2e209cea7d3524e1b537faaabc4ca1c6 (diff)
moving in the live stuff
Diffstat (limited to 'cli/app/search/search_class.py')
-rw-r--r--cli/app/search/search_class.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py
index 8825ca5..875bc75 100644
--- a/cli/app/search/search_class.py
+++ b/cli/app/search/search_class.py
@@ -191,9 +191,9 @@ def find_nearest_vector(sess, generator, opt_fp_in, opt_dims, out_images, out_la
print('iter: {}, loss: {}'.format(i, curr_loss))
if i > 0:
if opt_stochastic_clipping and (i % opt_clip_interval) == 0 and i < opt_steps * 0.45:
- sess.run(clip_latent, { clipped_alpha: i / opt_steps })
+ sess.run(clip_latent, { clipped_alpha: (i / opt_steps) ** 2 })
if opt_label_clipping and (i % opt_clip_interval) == 0:
- sess.run(clip_labels, { normalized_alpha: i / opt_steps })
+ sess.run(clip_labels, { normalized_alpha: (i / opt_steps) ** 2 })
if opt_video and opt_snapshot_interval != 0 and (i % opt_snapshot_interval) == 0:
phi_guess = sess.run(output)
guess_im = imgrid(imconvert_uint8(phi_guess), cols=1)