From 736e2d0b2e0ae23ecca4ae2457cb002395db75e5 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 11 Dec 2019 10:53:11 +0100 Subject: dont optimize encoding --- inversion/image_inversion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'inversion/image_inversion.py') diff --git a/inversion/image_inversion.py b/inversion/image_inversion.py index 2927eac..591f4f7 100644 --- a/inversion/image_inversion.py +++ b/inversion/image_inversion.py @@ -282,7 +282,7 @@ if params.decay_lr: params.inv_it / params.decay_n, 0.1, staircase=True) else: lrate = tf.constant(params.lr) -trained_params = [encoding] if params.fixed_z else [latent, encoding] +trained_params = [encoding] if params.fixed_z else [latent] optimizer = tf.train.AdamOptimizer(learning_rate=lrate, beta1=0.9, beta2=0.999) inv_train_op = optimizer.minimize(inv_loss, var_list=trained_params, global_step=inv_step) -- cgit v1.2.3-70-g09d2