From 7492971e966bd69b7451db8f756343c9462e7441 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 8 Jan 2020 01:36:22 +0100 Subject: getting latent search updates working... --- cli/app/search/search_class.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'cli/app/search/search_class.py') diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py index c59bc9b..a9e68e9 100644 --- a/cli/app/search/search_class.py +++ b/cli/app/search/search_class.py @@ -47,8 +47,9 @@ feature_layer_names = { def find_nearest_vector_for_images(paths, opt_dims, opt_steps, opt_video, opt_tag, opt_limit=-1, opt_stochastic_clipping=0, opt_label_clipping=0, opt_use_feature_detector=False, opt_feature_layers=[1,2,4,7], opt_snapshot_interval=20): - sess = tf.compat.v1.Session() tf.reset_default_graph() + sess = tf.compat.v1.Session() + print("Initializing generator...") generator = hub.Module('https://tfhub.dev/deepmind/biggan-512/2') fp_inverses = os.path.join(app_cfg.DIR_INVERSES, opt_tag) @@ -177,6 +178,8 @@ 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): + 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)) -- cgit v1.2.3-70-g09d2