diff options
| -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 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) |