From 7b742326d25a21c9c6f97766e0501ec0f5362099 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 21 Jan 2020 01:12:53 +0100 Subject: more aggressive stochastic clipping --- cli/app/search/search_class.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py index 0c3fd29..921e586 100644 --- a/cli/app/search/search_class.py +++ b/cli/app/search/search_class.py @@ -193,9 +193,9 @@ 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.45: - sess.run(clip_latent, { clipped_alpha: (i / opt_steps) ** 2 }) - if opt_label_clipping and (i % opt_clip_interval) == 0 and i < opt_steps * 0.75: + 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 }) + 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 }) if opt_video and opt_snapshot_interval != 0 and (i % opt_snapshot_interval) == 0: phi_guess = sess.run(output) -- cgit v1.2.3-70-g09d2