From f01105b36e5a8acf75d6a00f66ad9287d79f7575 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 11 Dec 2019 02:12:33 +0100 Subject: test small sample size --- cli/app/search/json.py | 8 ++++---- inversion/image_inversion.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/cli/app/search/json.py b/cli/app/search/json.py index 0f6c71c..1e470bd 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": 4, + "batch_size": 1, "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": 4, + "sample_size": 1, "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": 4, + "batch_size": 1, "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": 4, + "sample_size": 1, "dataset": "inverses/{}/dataset.encodings.dense.hdf5".format(tag), "save_progress": True, } diff --git a/inversion/image_inversion.py b/inversion/image_inversion.py index 04c033b..76638f9 100644 --- a/inversion/image_inversion.py +++ b/inversion/image_inversion.py @@ -355,7 +355,7 @@ for image_batch, label_batch in image_gen: # Save target. sess.run(target.assign(image_batch)) if COND_GAN: - sess.run(label.assign(one_hot(label_batch))) + sess.run(label.assign(label_batch)) # Initialize encodings to random values. if params.pre_trained_latent: -- cgit v1.2.3-70-g09d2