diff options
Diffstat (limited to 'cli/app/search/json.py')
| -rw-r--r-- | cli/app/search/json.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/cli/app/search/json.py b/cli/app/search/json.py index bf6bcb9..ced84df 100644 --- a/cli/app/search/json.py +++ b/cli/app/search/json.py @@ -55,9 +55,10 @@ def save_params_latent(fp_out_dir, tag): fp_out_fn = os.path.join(fp_out_dir, "params_latent.json") write_json(data, fp_out_fn) -def make_params_dense(tag): +def make_params_dense(tag, folder_id): return { "tag": tag, + "folder_id": folder_id, "decay_n": 2, "features": True, "clip": 1.0, @@ -100,10 +101,10 @@ def make_params_dense(tag): "max_batches": 0, } -def params_dense_dict(tag): - return ParamsDict(make_params_dense(tag)) +def params_dense_dict(tag, folder_id=59): + return ParamsDict(make_params_dense(tag, folder_id)) -def save_params_dense(fp_out_dir, tag): - data = make_params_dense(tag) +def save_params_dense(fp_out_dir, tag, folder_id=59): + data = make_params_dense(tag, folder_id) fp_out_fn = os.path.join(fp_out_dir, "params_dense.json") write_json(data, fp_out_fn) |