From 7ac93739a52a79c795f63862a99b111dd3a4b3ab Mon Sep 17 00:00:00 2001 From: Cameron Date: Tue, 11 Oct 2016 11:53:00 -0700 Subject: Add files via upload --- neural_style.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'neural_style.py') 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: -- cgit v1.2.3-70-g09d2