diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 02:21:36 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-11 02:21:36 +0100 |
| commit | 339f9ae16c26cfe598d0c019a98ef58885f756bc (patch) | |
| tree | ba4b9d64d2cefc8ca896b4faf73c70bc4bd94d43 /cli/app/search/json.py | |
| parent | f01105b36e5a8acf75d6a00f66ad9287d79f7575 (diff) | |
append extra images to fill batch
Diffstat (limited to 'cli/app/search/json.py')
| -rw-r--r-- | cli/app/search/json.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/cli/app/search/json.py b/cli/app/search/json.py index 1e470bd..0f6c71c 100644 --- a/cli/app/search/json.py +++ b/cli/app/search/json.py @@ -32,14 +32,14 @@ def save_params_latent(fp_out_dir, tag): "lambda_feat": 1.0, "init_gen_dist": False, "log_activation_layer": "Generator_2/GBlock/Relu:0", - "batch_size": 1, + "batch_size": 4, "fixed_z": False, "feature_extractor_output": "InceptionV3/Mixed_7a", "init_lo": -0.001, "lambda_mse": 1.0, "lambda_reg": 0.1, "dist_loss": False, - "sample_size": 1, + "sample_size": 4, "dataset": "inverses/{}/dataset.encodings.hdf5".format(tag), "save_progress": True, } @@ -77,14 +77,14 @@ def save_params_dense(fp_out_dir, tag): "lambda_feat": 1.0, "init_gen_dist": False, "log_activation_layer": "Generator_2/GBlock/Relu:0", - "batch_size": 1, + "batch_size": 4, "fixed_z": True, "feature_extractor_output": "InceptionV3/Mixed_7a", "init_lo": -0.001, "lambda_mse": 1.0, "lambda_reg": 0.1, "dist_loss": True, - "sample_size": 1, + "sample_size": 4, "dataset": "inverses/{}/dataset.encodings.dense.hdf5".format(tag), "save_progress": True, } |