diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-08 01:48:32 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-08 01:48:32 +0100 |
| commit | daa10de26027528528f485b6dfed94864faf259a (patch) | |
| tree | fdc88652a5d6ca53b2fd53a4d5ebf2e4257abc66 /cli/app/search/search_class.py | |
| parent | 7492971e966bd69b7451db8f756343c9462e7441 (diff) | |
up
Diffstat (limited to 'cli/app/search/search_class.py')
| -rw-r--r-- | cli/app/search/search_class.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py index a9e68e9..a4bec07 100644 --- a/cli/app/search/search_class.py +++ b/cli/app/search/search_class.py @@ -178,9 +178,9 @@ def find_nearest_vector(sess, generator, opt_fp_in, opt_dims, out_images, out_la try: print("Preparing to iterate...") for i in range(opt_steps): + curr_loss, _, _ = sess.run([loss, train_step_z, train_step_y], feed_dict=feed_dict) if i == 0: print("Iterating!") - 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 and (i % opt_stochastic_clipping) == 0: |