diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2019-12-15 01:48:57 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2019-12-15 01:48:57 +0100 |
| commit | a903afde8b8deb7dd44baaddf9a018c74f41336e (patch) | |
| tree | 4a25367aba9b7f4ae89fcac1fc5517d40337dad2 | |
| parent | fcf800545af9297fe9f1f70fb318f78ee6bdadfa (diff) | |
testing lower layers
| -rw-r--r-- | inversion/image_inversion_inception.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/inversion/image_inversion_inception.py b/inversion/image_inversion_inception.py index ae4793f..c11b94e 100644 --- a/inversion/image_inversion_inception.py +++ b/inversion/image_inversion_inception.py @@ -254,7 +254,7 @@ if params.features: gen_feat = gen_feat_ex["InceptionV3/Conv2d_1a_3x3"] target_feat = target_feat_ex["InceptionV3/Conv2d_1a_3x3"] feat_square_diff = tf.reshape(tf.square(gen_feat - target_feat), [BATCH_SIZE, -1]) - feat_loss += tf.reduce_mean(feat_square_diff) * 0.25 + 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_2a_3x3"] |