diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-07 22:23:41 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-07 22:23:41 +0100 |
| commit | 17b34ff357a9d015e87be5905f3dd8276c92b304 (patch) | |
| tree | 3d1b8fe74be8e407d5f6828a74e42a5d229a7ff7 /cli | |
| parent | 3be5df9a1939562248cf1740e9ed455680780403 (diff) | |
getting latent search updates working...
Diffstat (limited to 'cli')
| -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: |