diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-25 18:30:39 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-25 18:30:39 +0100 |
| commit | 176af09d1aa3caeecd454a8fe9754e5cb5dc75ff (patch) | |
| tree | cb184d2f8bb1015d1d23936c472e4e8343497e7b /cli/app/search/search_class.py | |
| parent | f6fda6b7ab25c8a3745997d7ea5eb3663e51955c (diff) | |
adjust
Diffstat (limited to 'cli/app/search/search_class.py')
| -rw-r--r-- | cli/app/search/search_class.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py index a98d7e3..371316e 100644 --- a/cli/app/search/search_class.py +++ b/cli/app/search/search_class.py @@ -200,12 +200,12 @@ def find_nearest_vector(sess, generator, opt_fp_in, opt_dims, out_images, out_la if i % 20 == 0: 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.75: - sess.run(clip_latent, { clipped_alpha: (i / opt_steps) }) + if opt_stochastic_clipping and (i % opt_clip_interval) == 0: # and i < opt_steps * 0.75: + sess.run(clip_latent, { clipped_alpha: 0.0 }) sess.run(reinit_optimizer_z) - if opt_label_clipping and (i % opt_clip_interval) == 0 and i < opt_steps * 0.75: + 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 }) sess.run(reinit_optimizer_y) if opt_video and opt_snapshot_interval != 0 and (i % opt_snapshot_interval) == 0: phi_guess = sess.run(output) |