From 6840bd72571b7e18b76f901891648fca5ff7352a Mon Sep 17 00:00:00 2001 From: cam Date: Mon, 27 Feb 2017 16:02:18 -0700 Subject: Updated to TensorFlow 1.0 --- neural_style.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/neural_style.py b/neural_style.py index b220f12..13ba513 100644 --- a/neural_style.py +++ b/neural_style.py @@ -382,8 +382,8 @@ def mask_style_layer(a, x, mask_img): mask = tf.stack(tensors, axis=2) mask = tf.stack(mask, axis=0) mask = tf.expand_dims(mask, 0) - a = tf.mul(a, mask) - x = tf.mul(x, mask) + a = tf.multiply(a, mask) + x = tf.multiply(x, mask) return a, x def sum_masked_style_losses(sess, net, style_imgs): -- cgit v1.2.3-70-g09d2