diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-20 01:49:04 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-20 01:49:04 +0100 |
| commit | 45a43ab57c827e673656f3e3ffb8f8acbcec4d22 (patch) | |
| tree | 075c2a726b7d16dc18b2ce727219c75275a5fb50 | |
| parent | 1eb65b5a3b99e6ae4465f08ddc9bbdc01077c4fa (diff) | |
iterate latent search longer
| -rw-r--r-- | cli/app/commands/biggan/extract_dense_vectors.py | 2 | ||||
| -rw-r--r-- | cli/app/search/json.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/commands/biggan/extract_dense_vectors.py b/cli/app/commands/biggan/extract_dense_vectors.py index 390bd85..8f5f8bb 100644 --- a/cli/app/commands/biggan/extract_dense_vectors.py +++ b/cli/app/commands/biggan/extract_dense_vectors.py @@ -10,7 +10,7 @@ from app.search.params import timestamp @click.command('') @click.option('-f', '--folder_id', 'opt_folder_id', type=int, help='Folder ID to process') -@click.option('-ls', '--latent_steps', 'opt_latent_steps', default=1000, type=int, +@click.option('-ls', '--latent_steps', 'opt_latent_steps', default=2000, type=int, help='Number of optimization iterations') @click.option('-ds', '--dense_steps', 'opt_dense_steps', default=2000, type=int, help='Number of optimization iterations') diff --git a/cli/app/search/json.py b/cli/app/search/json.py index c50d716..7d9f5fe 100644 --- a/cli/app/search/json.py +++ b/cli/app/search/json.py @@ -68,7 +68,7 @@ def make_params_dense(tag, folder_id): "dataset": os.path.join(app_cfg.DIR_INVERSES, tag, "dataset.latent.hdf5"), "inv_layer": "Generator_2/G_Z/Reshape:0", "decay_lr": False, - "inv_it": 5000, + "inv_it": 10000, "generator_path": "https://tfhub.dev/deepmind/biggan-512/2", "attention_map_layer": "Generator_2/attention/Softmax:0", "pre_trained_latent": True, |