diff options
| -rw-r--r-- | cli/app/commands/biggan/extract_dense_vectors.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/commands/biggan/extract_dense_vectors.py b/cli/app/commands/biggan/extract_dense_vectors.py index 754dbd3..8bcd973 100644 --- a/cli/app/commands/biggan/extract_dense_vectors.py +++ b/cli/app/commands/biggan/extract_dense_vectors.py @@ -10,7 +10,7 @@ from app.search.params import timestamp @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=1000, type=int, +@click.option('-ls', '--latent_steps', 'opt_latent_steps', default=2000, type=int, help='Number of optimization iterations') @click.option('-ds', '--dense_steps', 'opt_dense_steps', default=2000, type=int, help='Number of optimization iterations') @@ -57,7 +57,7 @@ def cli(ctx, opt_folder_id, opt_latent_steps, opt_dense_steps, opt_video, opt_re paths=paths, opt_dims=512, opt_steps=opt_latent_steps, - opt_video=opt_video, + opt_video=True, opt_tag=tag, opt_limit=-1, opt_stochastic_clipping=opt_stochastic_clipping, |