diff options
| -rw-r--r-- | inversion/image_inversion.py | 2 |
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) |