From 27dc60cee28e1406ca213931ee80ace19efb758a Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Thu, 20 Feb 2020 23:44:19 +0100 Subject: vgg? --- cli/app/search/search_class.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'cli/app/search/search_class.py') 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 -- cgit v1.2.3-70-g09d2