diff options
| -rw-r--r-- | cli/app/search/search_class.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py index 42ff605..8937eec 100644 --- a/cli/app/search/search_class.py +++ b/cli/app/search/search_class.py @@ -141,9 +141,9 @@ def find_nearest_vector(sess, generator, opt_fp_in, opt_dims, out_images, out_la # Batch reconstruction error. inv_loss = 1.0 * mse_loss + 1.0 * feat_loss - train_step_z = tf.train.AdamOptimizer(learning_rate=z_lr, beta1=0.9, beta2=0.999) + train_step_z = tf.train.AdamOptimizer(learning_rate=z_lr, beta1=0.9, beta2=0.999) \ .minimize(inv_loss, var_list=[input_z]) - train_step_y = tf.train.AdamOptimizer(learning_rate=y_lr, beta1=0.9, beta2=0.999) + train_step_y = tf.train.AdamOptimizer(learning_rate=y_lr, beta1=0.9, beta2=0.999) \ .minimize(inv_loss, var_list=[input_y]) else: |