summaryrefslogtreecommitdiff
path: root/cli/app/commands/biggan/extract_dense_vectors.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-01-21 14:42:02 +0100
committerJules Laplace <julescarbon@gmail.com>2020-01-21 14:42:02 +0100
commitb503a0b47c5f85e380fa4ab24deed2420456ae5f (patch)
treedbb02bce3f81ec1a2e4dd396afdca36d8efc5a71 /cli/app/commands/biggan/extract_dense_vectors.py
parent7b742326d25a21c9c6f97766e0501ec0f5362099 (diff)
emphasize matching class rather than minimizing l1
Diffstat (limited to 'cli/app/commands/biggan/extract_dense_vectors.py')
-rw-r--r--cli/app/commands/biggan/extract_dense_vectors.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/commands/biggan/extract_dense_vectors.py b/cli/app/commands/biggan/extract_dense_vectors.py
index f7cffb2..3ca7313 100644
--- a/cli/app/commands/biggan/extract_dense_vectors.py
+++ b/cli/app/commands/biggan/extract_dense_vectors.py
@@ -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,3a,7a",
help='Feature layers used for loss')
@click.option('-snap', '--snapshot_interval', 'opt_snapshot_interval', default=20,
help='Interval to store sample images')