summaryrefslogtreecommitdiff
path: root/cli/app/commands
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-01-06 14:30:51 +0100
committerJules Laplace <julescarbon@gmail.com>2020-01-06 14:30:51 +0100
commita11656ea3cb3af5ad58997b2ad8d5e94677284dc (patch)
tree7cf32395172b909b085741efb57d5827885b121e /cli/app/commands
parent4f01dc2d69a17281d32c2f36f03285d910783675 (diff)
path
Diffstat (limited to 'cli/app/commands')
-rw-r--r--cli/app/commands/biggan/search_class.py2
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 4c797ae..28a01fd 100644
--- a/cli/app/commands/biggan/search_class.py
+++ b/cli/app/commands/biggan/search_class.py
@@ -66,7 +66,7 @@ def cli(ctx, opt_fp_in, opt_dims, opt_steps, opt_limit, opt_video, opt_tag):
if index == opt_limit:
break
out_fns[index] = os.path.basename(path)
- fp_frames = find_nearest_vector(generator, sess, path, opt_dims, out_images, out_labels, opt_steps, index)
+ fp_frames = find_nearest_vector(sess, path, opt_dims, out_images, out_labels, opt_steps, index)
if opt_video:
export_video(fp_frames)