diff options
| author | cam <cameron@ideum.com> | 2016-10-15 16:36:24 -0600 |
|---|---|---|
| committer | cam <cameron@ideum.com> | 2016-10-15 16:36:24 -0600 |
| commit | 5ca01d831a0ceec853135b9eb35e65ed8e327d82 (patch) | |
| tree | 98f6eb0ca54b12cd571039729e78dd7e9f3f10aa | |
| parent | 6ebf3429d2279bd0fddbacf1d674c6f1f04cea04 (diff) | |
Fixed bug between tensorflow builds 0.9, 0.10rc, 0.10
| -rw-r--r-- | neural_style.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/neural_style.py b/neural_style.py index d0bb45f..1dd0c6c 100644 --- a/neural_style.py +++ b/neural_style.py @@ -358,7 +358,7 @@ def style_layer_loss(a, x): return loss def gram_matrix(x, area, depth): - F = tf.reshape(x[0], (area, depth)) + F = tf.reshape(x, (area, depth)) G = tf.matmul(tf.transpose(F), F) return G |