summaryrefslogtreecommitdiff
path: root/cli/app/commands/biggan
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-01-20 17:33:30 +0100
committerJules Laplace <julescarbon@gmail.com>2020-01-20 17:33:30 +0100
commit47e4c11b8c79ba737cc538af47b9a7caebaac18b (patch)
tree3004d3f426582d8601a99a95b88abcb16a233f3d /cli/app/commands/biggan
parent45a43ab57c827e673656f3e3ffb8f8acbcec4d22 (diff)
more layers
Diffstat (limited to 'cli/app/commands/biggan')
-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 8f5f8bb..f7cffb2 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="2a,4a,7a",
+@click.option('-ll', '--feature_layers', 'opt_feature_layers', default="1a,2a,3a,4a,7a",
help='Feature layers used for loss')
@click.option('-snap', '--snapshot_interval', 'opt_snapshot_interval', default=20,
help='Interval to store sample images')