diff options
| author | cam <cameron@ideum.com> | 2017-02-27 16:02:18 -0700 |
|---|---|---|
| committer | cam <cameron@ideum.com> | 2017-02-27 16:02:18 -0700 |
| commit | 6840bd72571b7e18b76f901891648fca5ff7352a (patch) | |
| tree | b95124bfa76017e15702458e6ef5afd157b4696b | |
| parent | aec13c76fe224daafaf534b8c5a26d772b1a4db5 (diff) | |
Updated to TensorFlow 1.0
| -rw-r--r-- | neural_style.py | 4 |
1 files 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): |