diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-09 10:22:54 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-09 10:22:54 +0100 |
| commit | c7ad87acbce1b307b49489eb11a6f5f8740a66e3 (patch) | |
| tree | ae9983ead61c5568c0e3e2dc9733f206325772e0 /cli/app/commands | |
| parent | f8b1bce9df80239a523832397fca2a0f3e5c500d (diff) | |
dont normalize
Diffstat (limited to 'cli/app/commands')
| -rw-r--r-- | cli/app/commands/biggan/search.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/cli/app/commands/biggan/search.py b/cli/app/commands/biggan/search.py index aeda9f5..e764487 100644 --- a/cli/app/commands/biggan/search.py +++ b/cli/app/commands/biggan/search.py @@ -184,7 +184,7 @@ def find_nearest_vector(generator, sess, input_z, input_y, input_trunc, output, imwrite(join(app_cfg.DIR_OUTPUTS, fp_frames, 'frame_0000_start.png'), start_im) try: - for i in range(1000): + for i in range(500): feed_dict = {input_z: z, input_y: y, input_trunc: truncation} grad_z, grad_y = sess.run([dc_dz, dc_dy], feed_dict=feed_dict) z -= grad_z * lr_z @@ -194,10 +194,10 @@ def find_nearest_vector(generator, sess, input_z, input_y, input_trunc, output, lr_y *= 0.999 if i % 30 == 0: - lr_y *= 1.002 + # lr_y *= 1.002 y = np.clip(y, 0, 1) - for j in range(batch_size): - y[j] /= y[j].sum() + # for j in range(batch_size): + # y[j] /= y[j].sum() if i > 200 and i % 100 == 0: mean = np.mean(y, axis=0) y = y * 3 / 4 + mean / 4 |