diff options
| author | Cameron <cysmith1010@gmail.com> | 2016-10-11 11:53:00 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2016-10-11 11:53:00 -0700 |
| commit | 7ac93739a52a79c795f63862a99b111dd3a4b3ab (patch) | |
| tree | 488f2bfad41da1359fc9dfc3b1284c3c1879854e /neural_style.py | |
| parent | e32e041c61878464a7270a6050349dbfcdba10e7 (diff) | |
Add files via upload
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: |