diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-08 17:54:36 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-08 17:54:36 +0100 |
| commit | 8fc7f19a632408a59134ecc092d21d6a31fbb723 (patch) | |
| tree | 6f06d10c99f2a083f07f34d306a584b4256c5989 /cli/app/search/search_dense.py | |
| parent | 034922d32c1d9df996e6292a17fb4fb4cb04395d (diff) | |
more latent tweaking
Diffstat (limited to 'cli/app/search/search_dense.py')
| -rw-r--r-- | cli/app/search/search_dense.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py index 362e0ce..d5cbf64 100644 --- a/cli/app/search/search_dense.py +++ b/cli/app/search/search_dense.py @@ -246,8 +246,8 @@ def find_dense_embedding_for_images(params): # feat_loss += tf.reduce_mean(feat_square_diff) * 0.25 # img_feat_err += tf.reduce_mean(feat_square_diff, axis=1) * 0.25 - gen_feat = gen_feat_ex["InceptionV3/Conv2d_4a_3x3"] - target_feat = target_feat_ex["InceptionV3/Conv2d_4a_3x3"] + gen_feat = gen_feat_ex["InceptionV3/Mixed_6a"] + target_feat = target_feat_ex["InceptionV3/Mixed_6a"] feat_square_diff = tf.reshape(tf.square(gen_feat - target_feat), [BATCH_SIZE, -1]) feat_loss += tf.reduce_mean(feat_square_diff) * 0.25 img_feat_err += tf.reduce_mean(feat_square_diff, axis=1) * 0.25 |