diff options
Diffstat (limited to 'cli/app/commands/biggan/extract_dense_vectors.py')
| -rw-r--r-- | cli/app/commands/biggan/extract_dense_vectors.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/cli/app/commands/biggan/extract_dense_vectors.py b/cli/app/commands/biggan/extract_dense_vectors.py index 4e7952a..b4300b5 100644 --- a/cli/app/commands/biggan/extract_dense_vectors.py +++ b/cli/app/commands/biggan/extract_dense_vectors.py @@ -9,9 +9,9 @@ from app.search.json import params_dense_dict @click.command('') @click.option('-f', '--folder_id', 'opt_folder_id', type=int, help='Folder ID to process') -@click.option('-ls', '--latent_steps', 'opt_latent_steps', default=2000, type=int, +@click.option('-ls', '--latent_steps', 'opt_latent_steps', default=1000, type=int, help='Number of optimization iterations') -@click.option('-ds', '--dense_steps', 'opt_dense_steps', default=2000, type=int, +@click.option('-ds', '--dense_steps', 'opt_dense_steps', default=1000, type=int, help='Number of optimization iterations') @click.option('-v', '--video', 'opt_video', is_flag=True, help='Export a video for each dataset') @@ -55,7 +55,7 @@ def cli(ctx, opt_folder_id, opt_latent_steps, opt_dense_steps, opt_video, opt_re find_nearest_vector_for_images( paths=paths, opt_dims=512, - opt_steps=opt_dense_steps, + opt_steps=opt_latent_steps, opt_video=opt_video, opt_tag=tag, opt_limit=-1, |