diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-25 18:30:39 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-25 18:30:39 +0100 |
| commit | 176af09d1aa3caeecd454a8fe9754e5cb5dc75ff (patch) | |
| tree | cb184d2f8bb1015d1d23936c472e4e8343497e7b /cli/app/commands/biggan/extract_dense_vectors.py | |
| parent | f6fda6b7ab25c8a3745997d7ea5eb3663e51955c (diff) | |
adjust
Diffstat (limited to 'cli/app/commands/biggan/extract_dense_vectors.py')
| -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 e8f0587..eb811d5 100644 --- a/cli/app/commands/biggan/extract_dense_vectors.py +++ b/cli/app/commands/biggan/extract_dense_vectors.py @@ -1,4 +1,4 @@ -import click +2import click import os from app.utils.cortex_utils import cortex_folder, download_cortex_files, find_unprocessed_files @@ -24,7 +24,7 @@ from app.search.params import timestamp 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,3a,4a,7a", +@click.option('-ll', '--feature_layers', 'opt_feature_layers', default="1a,4a,7a", help='Feature layers used for loss') @click.option('-snap', '--snapshot_interval', 'opt_snapshot_interval', default=20, help='Interval to store sample images') |