Minor fix for readability.

herrlustig · Apr 21, 2016 · 2d98303 · 2d98303
1 parent 7e96851
commit 2d98303
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doodle.py b/doodle.py
@@ -415,7 +415,7 @@ def style_loss(self):
             dist = self.model.tensor_outputs['nn'+l]
             dist = dist.reshape((dist.shape[1], -1)) / norms.reshape((1,-1)) / layer.N.reshape((-1,1))
 
-            if args.semantic_weight:
+            if args.semantic_weight > 0.0:
                 sem_layer = self.model.network['mm'+l]
                 sem = self.model.tensor_outputs['mm'+l]
                 sem = sem.reshape((sem.shape[1], -1)) / sem_norms.reshape((1,-1)) / sem_layer.N.reshape((-1,1))