From 91ac0c68b63639ab545b35ad98aa31e4783dfef1 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 16 Feb 2020 11:01:40 +0100 Subject: scope --- cli/app/search/search_dense.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'cli') diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py index bb6f636..18c430e 100644 --- a/cli/app/search/search_dense.py +++ b/cli/app/search/search_dense.py @@ -567,7 +567,7 @@ def feature_loss_tfhub(feature_extractor, opt_feature_layers, BATCH_SIZE, img_a, return feat_loss / len(opt_feature_layers), img_feat_err / len(opt_feature_layers) scope_index = 0 -vgg_model = tf.make_template(nets.vgg.vgg_16) +vgg_model = tf.make_template('vgg16', nets.vgg.vgg_16, is_training=False) def feature_loss_vgg(feature_extractor, opt_feature_layers, BATCH_SIZE, img_a, img_b, y, x, height, width, resize_height=None, resize_width=None): height = int(height) width = int(width) -- cgit v1.2.3-70-g09d2