summaryrefslogtreecommitdiff
path: root/cli/app/search/search_class.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-02-20 23:44:19 +0100
committerJules Laplace <julescarbon@gmail.com>2020-02-20 23:44:19 +0100
commit27dc60cee28e1406ca213931ee80ace19efb758a (patch)
treea03215f1804bcf13c72950cbdaefeafea5a395cb /cli/app/search/search_class.py
parent3037d8d8febb7a773d786312c557a98b8d21bd39 (diff)
vgg?
Diffstat (limited to 'cli/app/search/search_class.py')
-rw-r--r--cli/app/search/search_class.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py
index 3c44fdb..4336c0a 100644
--- a/cli/app/search/search_class.py
+++ b/cli/app/search/search_class.py
@@ -174,7 +174,7 @@ def find_nearest_vector(sess, generator, opt_fp_in, opt_dims, out_images, out_la
feat_loss_vgg, img_feat_err_vgg = feature_loss_vgg(None, opt_feature_layers, batch_size, gen_img_ch, target_img_ch, None, None, height, width)
- loss = 10.0 * mse_loss + 1.0 * feat_loss_vgg
+ loss = 1.0 * mse_loss + 1.0 * feat_loss_vgg
z_lr = 0.001
y_lr = 0.001