diff options
| -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') |