diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-16 02:22:03 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-16 02:22:03 +0100 |
| commit | 6e96d6198f5a7726e135ebae5228646ca8a22f2e (patch) | |
| tree | 738cc489c22cf8b839808380290400bcc396b25f /cli/app/commands/biggan | |
| parent | ad434493fe0203e632d7e7abeae0ce339d0a26bb (diff) | |
less crazy iterations... and sorting
Diffstat (limited to 'cli/app/commands/biggan')
| -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, |