From 1fc1e2c4922e8d47ff0bcdce08e3e2d1850f3d44 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 11 Feb 2020 19:06:37 +0100 Subject: try g(z) encoding and label embedding --- cli/app/search/json.py | 4 ++-- cli/app/search/search_dense.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/cli/app/search/json.py b/cli/app/search/json.py index 35a9daf..83893c8 100644 --- a/cli/app/search/json.py +++ b/cli/app/search/json.py @@ -67,8 +67,8 @@ def make_params_dense(tag, folder_id): "clipping": False, "path": os.path.join(app_cfg.DIR_INVERSES, tag), "dataset": os.path.join(app_cfg.DIR_INVERSES, tag, "dataset.latent.hdf5"), - # "inv_layer": "Generator_2/G_Z/Reshape:0", - "inv_layer": "latent", + "inv_layer": "Generator_2/G_Z/Reshape:0", + # "inv_layer": "latent", "decay_lr": True, "inv_it": 10000, "generator_path": "https://tfhub.dev/deepmind/biggan-512/2", diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py index 458d5b8..4d59da9 100644 --- a/cli/app/search/search_dense.py +++ b/cli/app/search/search_dense.py @@ -227,7 +227,7 @@ def find_dense_embedding_for_images(params, opt_tag="inverse_" + timestamp(), op img_rec_err = params.lambda_mse * img_mse_err + params.lambda_feat * img_feat_err inv_loss = params.lambda_mse * mse_loss + params.lambda_feat * feat_loss - inv_loss_quad = params.lambda_mse * mse_loss + params.lambda_feat * feat_loss_quad + inv_loss_quad = 5.0 * mse_loss + params.lambda_feat * feat_loss_quad # -------------------------- # Optimizer. -- cgit v1.2.3-70-g09d2