diff options
Diffstat (limited to 'cli')
| -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: |