From 92ddbd73d20e3292de517bbfaaf66078c4d47314 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 15 Dec 2019 01:53:14 +0100 Subject: testing lower layers --- inversion/image_inversion_inception.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'inversion/image_inversion_inception.py') diff --git a/inversion/image_inversion_inception.py b/inversion/image_inversion_inception.py index c11b94e..570c1ac 100644 --- a/inversion/image_inversion_inception.py +++ b/inversion/image_inversion_inception.py @@ -255,7 +255,7 @@ if params.features: 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 - img_feat_err += tf.reduce_mean(feat_square_diff, axis=1) * 0.25 + img_feat_err = tf.reduce_mean(feat_square_diff, axis=1) * 0.25 gen_feat = gen_feat_ex["InceptionV3/Conv2d_2a_3x3"] target_feat = target_feat_ex["InceptionV3/Conv2d_2a_3x3"] -- cgit v1.2.3-70-g09d2