diff options
Diffstat (limited to 'cli/app/commands/biggan/extract_dense_vectors.py')
| -rw-r--r-- | cli/app/commands/biggan/extract_dense_vectors.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/cli/app/commands/biggan/extract_dense_vectors.py b/cli/app/commands/biggan/extract_dense_vectors.py index 9f7644e..5320496 100644 --- a/cli/app/commands/biggan/extract_dense_vectors.py +++ b/cli/app/commands/biggan/extract_dense_vectors.py @@ -25,9 +25,11 @@ from app.search.json import params_dense_dict help='Feature layers used for loss') @click.option('-snap', '--snapshot_interval', 'opt_snapshot_interval', default=20, help='Interval to store sample images') +@click.option('-clip', '--clip_interval', 'opt_clip_interval', default=500, + help='Interval to clip vectors') @click.pass_context def cli(ctx, opt_folder_id, opt_latent_steps, opt_dense_steps, opt_video, - opt_stochastic_clipping, opt_label_clipping, opt_use_feature_detector, opt_feature_layers, opt_snapshot_interval): + opt_stochastic_clipping, opt_label_clipping, opt_use_feature_detector, opt_feature_layers, opt_snapshot_interval, opt_clip_interval): """ The full process: - Fetch new images from the cortex @@ -59,7 +61,8 @@ def cli(ctx, opt_folder_id, opt_latent_steps, opt_dense_steps, opt_video, opt_label_clipping=opt_label_clipping, opt_use_feature_detector=opt_use_feature_detector, opt_feature_layers=opt_feature_layers, - opt_snapshot_interval=opt_snapshot_interval + opt_snapshot_interval=opt_snapshot_interval, + opt_clip_interval=opt_clip_interval ) params = params_dense_dict(tag) |