From daa10de26027528528f485b6dfed94864faf259a Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 8 Jan 2020 01:48:32 +0100 Subject: up --- cli/app/search/search_class.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: -- cgit v1.2.3-70-g09d2