diff options
Diffstat (limited to 'cli/app/commands/biggan')
| -rw-r--r-- | cli/app/commands/biggan/extract_dense_vectors.py | 4 | ||||
| -rw-r--r-- | cli/app/commands/biggan/search_class.py | 8 |
2 files changed, 6 insertions, 6 deletions
diff --git a/cli/app/commands/biggan/extract_dense_vectors.py b/cli/app/commands/biggan/extract_dense_vectors.py index 7fef459..45a13db 100644 --- a/cli/app/commands/biggan/extract_dense_vectors.py +++ b/cli/app/commands/biggan/extract_dense_vectors.py @@ -17,9 +17,9 @@ from app.search.json import params_dense_dict help='Export a video for each dataset') @click.option('-rp', '--reprocess', 'opt_reprocess', is_flag=True, help='Reprocess images') -@click.option('-sc', '--stochastic_clipping', 'opt_stochastic_clipping', default=0, +@click.option('-sc', '--stochastic_clipping', 'opt_stochastic_clipping', is_flag=True, help='Compute feature loss') -@click.option('-lc', '--label_clipping', 'opt_label_clipping', default=0, +@click.option('-lc', '--label_clipping', 'opt_label_clipping', is_flag=True, help='Normalize labels every N steps') @click.option('-feat', '--use_feature_detector', 'opt_use_feature_detector', is_flag=True, help='Compute feature loss') diff --git a/cli/app/commands/biggan/search_class.py b/cli/app/commands/biggan/search_class.py index 050fbef..7a7bfdb 100644 --- a/cli/app/commands/biggan/search_class.py +++ b/cli/app/commands/biggan/search_class.py @@ -19,10 +19,10 @@ from app.utils.cortex_utils import cortex_folder, download_cortex_files, find_un help='Export a video for each dataset') @click.option('-t', '--tag', 'opt_tag', default='inverse_' + str(int(time.time() * 1000)), help='Tag this dataset') -@click.option('-sc', '--stochastic_clipping', 'opt_stochastic_clipping', default=0, - help='Compute feature loss. Specify interval to do the clipping') -@click.option('-lc', '--label_clipping', 'opt_label_clipping', default=0, - help='Normalize labels every N steps. Specify interval to do the normalization') +@click.option('-sc', '--stochastic_clipping', 'opt_stochastic_clipping', is_flag=True, + help='Compute feature loss') +@click.option('-lc', '--label_clipping', 'opt_label_clipping', is_flag=True, + 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", |