diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-02-21 00:05:51 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-02-21 00:05:51 +0100 |
| commit | 6ded7524ba2e872c35f5789159e36acf8a096813 (patch) | |
| tree | 843f2aa2600992c2975b1d939db48c2dff827ce8 /cli | |
| parent | 945e67e02b0e75a0a179ddcc03c72f0a12b82193 (diff) | |
vgg?
Diffstat (limited to 'cli')
| -rw-r--r-- | cli/app/search/search_class.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/search/search_class.py b/cli/app/search/search_class.py index 4336c0a..d88f51c 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 = 1.0 * mse_loss + 1.0 * feat_loss_vgg + loss = 1.0 * mse_loss + 10.0 * feat_loss_vgg z_lr = 0.001 y_lr = 0.001 |