From 6e96d6198f5a7726e135ebae5228646ca8a22f2e Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 16 Jan 2020 02:22:03 +0100 Subject: less crazy iterations... and sorting --- cli/app/commands/biggan/extract_dense_vectors.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'cli/app/commands/biggan') 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, -- cgit v1.2.3-70-g09d2