From 2b93907f97eea42dfd444c070509488dabbfc2ff Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 7 Jan 2020 22:21:45 +0100 Subject: getting latent search updates working... --- cli/app/commands/biggan/extract_dense_vectors.py | 1 + cli/app/commands/biggan/search_class.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) (limited to 'cli/app/commands') diff --git a/cli/app/commands/biggan/extract_dense_vectors.py b/cli/app/commands/biggan/extract_dense_vectors.py index 7dc25bc..85e751c 100644 --- a/cli/app/commands/biggan/extract_dense_vectors.py +++ b/cli/app/commands/biggan/extract_dense_vectors.py @@ -46,6 +46,7 @@ def cli(ctx, opt_folder_id, opt_latent_steps, opt_dense_steps, opt_video, tag = "folder_{}".format(folder['id']) paths = [file['path'] for file in unprocessed_files] + opt_feature_layers = opt_feature_layers.split(',') find_nearest_vector_for_images( paths=paths, diff --git a/cli/app/commands/biggan/search_class.py b/cli/app/commands/biggan/search_class.py index 1f12080..f307054 100644 --- a/cli/app/commands/biggan/search_class.py +++ b/cli/app/commands/biggan/search_class.py @@ -25,7 +25,7 @@ from app.search.search_class import find_nearest_vector_for_images 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" +@click.option('-ll', '--feature_layers', 'opt_feature_layers', default="1a,2a,4a,7a", help='Feature layers used for loss') @click.option('-snap', '--snapshot_interval', 'opt_snapshot_interval', default=20 help='Interval to store sample images') -- cgit v1.2.3-70-g09d2