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.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/cli/app/commands/biggan/extract_dense_vectors.py b/cli/app/commands/biggan/extract_dense_vectors.py
index 866104d..4e7952a 100644
--- a/cli/app/commands/biggan/extract_dense_vectors.py
+++ b/cli/app/commands/biggan/extract_dense_vectors.py
@@ -69,4 +69,8 @@ def cli(ctx, opt_folder_id, opt_latent_steps, opt_dense_steps, opt_video, opt_re
)
params = params_dense_dict(tag, folder_id=folder['id'])
- find_dense_embedding_for_images(params)
+ find_dense_embedding_for_images(params,
+ opt_tag=tag,
+ opt_feature_layers=opt_feature_layers,
+ opt_save_progress=False)
+