diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-08 01:17:08 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-08 01:17:08 +0100 |
| commit | 9f0c0691a4f0c1e54d7216e4596df0ea4b8c7226 (patch) | |
| tree | 22bf26560e2bc29e6d438f85acfbdb074c53843e /cli/app/search | |
| parent | c00bc938019f660b72fff7f9972f93eb7462d8cf (diff) | |
getting latent search updates working...
Diffstat (limited to 'cli/app/search')
| -rw-r--r-- | cli/app/search/json.py | 2 | ||||
| -rw-r--r-- | cli/app/search/search_class.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/search/json.py b/cli/app/search/json.py index b47512d..4403353 100644 --- a/cli/app/search/json.py +++ b/cli/app/search/json.py @@ -3,7 +3,7 @@ from app.settings import app_cfg from app.utils.file_utils import write_json from app.search.params import ParamsDict -def make_params_latent(fp_out_dir, tag): +def make_params_latent(tag): return { "tag": tag, "decay_n": 2, diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py index 2296cb9..f13e1fd 100644 --- a/cli/app/search/search_class.py +++ b/cli/app/search/search_class.py @@ -53,7 +53,7 @@ def find_nearest_vector_for_images(paths, opt_dims, opt_steps, opt_video, opt_ta fp_inverses = os.path.join(app_cfg.DIR_INVERSES, opt_tag) os.makedirs(fp_inverses, exist_ok=True) - save_params_latent(fp_inverses, opt_tag) + # save_params_latent(fp_inverses, opt_tag) save_params_dense(fp_inverses, opt_tag) out_file = h5py.File(join(fp_inverses, 'dataset.latent.hdf5'), 'w') out_images = out_file.create_dataset('xtrain', (len(paths), 3, 512, 512,), dtype='float32') |