dont optimize encoding

1 files changed, 1 insertions, 1 deletions

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)