diff options
| -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 7a305b0..667afae 100644 --- a/cli/app/search/search_class.py +++ b/cli/app/search/search_class.py @@ -194,10 +194,10 @@ 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) * 2 }) + sess.run(clip_latent, { clipped_alpha: 0.0 }) if opt_label_clipping and (i % opt_clip_interval) == 0: # and i < opt_steps * 0.75: # sess.run(clip_labels, { normalized_alpha: (i / opt_steps) ** 2 }) - sess.run(clip_labels, { normalized_alpha: i / opt_steps }) + sess.run(clip_labels, { normalized_alpha: 0.0 }) 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) |