summaryrefslogtreecommitdiff
path: root/cli/app/commands/biggan/extract_dense_vectors.py
diff options
context:
space:
mode:
Diffstat (limited to 'cli/app/commands/biggan/extract_dense_vectors.py')
-rw-r--r--cli/app/commands/biggan/extract_dense_vectors.py4
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 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')