diff options
Diffstat (limited to 'cli/app/commands/biggan/search_class.py')
| -rw-r--r-- | cli/app/commands/biggan/search_class.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/commands/biggan/search_class.py b/cli/app/commands/biggan/search_class.py index b681e1f..ab040c0 100644 --- a/cli/app/commands/biggan/search_class.py +++ b/cli/app/commands/biggan/search_class.py @@ -151,7 +151,7 @@ def find_nearest_vector(sess, generator, opt_fp_in, opt_dims, out_images, out_la except KeyboardInterrupt: pass - z_guess, y_guess = sess.run(input_z, input_y) + z_guess, y_guess = sess.run([input_z, input_y]) out_images[index] = phi_target_for_inversion out_labels[index] = y_guess out_latent[index] = z_guess |