diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-07 22:24:28 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-07 22:24:28 +0100 |
| commit | 863985b867f05bc716a8440f1abaa3c8c75ef71a (patch) | |
| tree | e78e74a32063e1b20b70e5d4e2e4e297e3fba869 /cli | |
| parent | 17b34ff357a9d015e87be5905f3dd8276c92b304 (diff) | |
getting latent search updates working...
Diffstat (limited to 'cli')
| -rw-r--r-- | cli/app/search/search_class.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py index 8937eec..7db518b 100644 --- a/cli/app/search/search_class.py +++ b/cli/app/search/search_class.py @@ -177,11 +177,11 @@ def find_nearest_vector(sess, generator, opt_fp_in, opt_dims, out_images, out_la curr_loss, _, _ = sess.run([loss, train_step_z, train_step_y], feed_dict=feed_dict) if i % 20 == 0: print('iter: {}, loss: {}'.format(i, curr_loss)) - if opt_stochastic_clipping != 0 && i % opt_stochastic_clipping == 0: + if opt_stochastic_clipping != 0 && (i % opt_stochastic_clipping) == 0: sess.run(clip_latent) - if opt_label_clipping != 0 && i % opt_label_clipping == 0: + if opt_label_clipping != 0 && (i % opt_label_clipping) == 0: sess.run(clip_labels) - if opt_snapshot_interval != 0 && i % opt_snapshot_interval == 0: + if opt_snapshot_interval != 0 && (i % opt_snapshot_interval) == 0: phi_guess = sess.run(output) guess_im = imgrid(imconvert_uint8(phi_guess), cols=1) imwrite(join(app_cfg.DIR_OUTPUTS, fp_frames, 'frame_{:04d}.png'.format(int(i / opt_snapshot_interval))), guess_im) |