diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-07 22:22:19 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-07 22:22:19 +0100 |
| commit | 3be5df9a1939562248cf1740e9ed455680780403 (patch) | |
| tree | ddfe0e761796a9719e8da2aa7374b220cd759c40 | |
| parent | 2b93907f97eea42dfd444c070509488dabbfc2ff (diff) | |
getting latent search updates working...
| -rw-r--r-- | cli/app/commands/biggan/extract_dense_vectors.py | 6 | ||||
| -rw-r--r-- | cli/app/commands/biggan/search_class.py | 2 |
2 files changed, 4 insertions, 4 deletions
diff --git a/cli/app/commands/biggan/extract_dense_vectors.py b/cli/app/commands/biggan/extract_dense_vectors.py index 85e751c..1f7f8c3 100644 --- a/cli/app/commands/biggan/extract_dense_vectors.py +++ b/cli/app/commands/biggan/extract_dense_vectors.py @@ -15,15 +15,15 @@ from app.search.json import params_dense_dict help='Number of optimization iterations') @click.option('-v', '--video', 'opt_video', is_flag=True, help='Export a video for each dataset') -@click.option('-sc', '--stochastic_clipping', 'opt_stochastic_clipping', default=0 +@click.option('-sc', '--stochastic_clipping', 'opt_stochastic_clipping', default=0, help='Compute feature loss') @click.option('-lc', '--label_clipping', 'opt_label_clipping', default=0, help='Normalize labels every N steps') @click.option('-feat', '--use_feature_detector', 'opt_use_feature_detector', is_flag=True, help='Compute feature loss') -@click.option('-ll', '--feature_layers', 'opt_feature_layers', default="1a,2a,4a,7a" +@click.option('-ll', '--feature_layers', 'opt_feature_layers', default="1a,2a,4a,7a", help='Feature layers used for loss') -@click.option('-snap', '--snapshot_interval', 'opt_snapshot_interval', default=20 +@click.option('-snap', '--snapshot_interval', 'opt_snapshot_interval', default=20, help='Interval to store sample images') @click.pass_context def cli(ctx, opt_folder_id, opt_latent_steps, opt_dense_steps, opt_video, diff --git a/cli/app/commands/biggan/search_class.py b/cli/app/commands/biggan/search_class.py index f307054..ca499a3 100644 --- a/cli/app/commands/biggan/search_class.py +++ b/cli/app/commands/biggan/search_class.py @@ -27,7 +27,7 @@ from app.search.search_class import find_nearest_vector_for_images help='Compute feature loss') @click.option('-ll', '--feature_layers', 'opt_feature_layers', default="1a,2a,4a,7a", help='Feature layers used for loss') -@click.option('-snap', '--snapshot_interval', 'opt_snapshot_interval', default=20 +@click.option('-snap', '--snapshot_interval', 'opt_snapshot_interval', default=20, help='Interval to store sample images') @click.pass_context |