diff options
Diffstat (limited to 'neural_style.py')
| -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 374b4c1..bac5222 100644 --- a/neural_style.py +++ b/neural_style.py @@ -340,7 +340,7 @@ def content_layer_loss(p, x): M = h.value * w.value N = d.value if args.content_loss_function == 1: - K = 1. / (2 * N**0.5 * M**0.5) + K = 1. / (2. * N**0.5 * M**0.5) elif args.content_loss_function == 2: K = 1. / (N * M) elif args.content_loss_function == 3: |