Merge pull request #32 from DmitryUlyanov/patch-1

Fix default parameter in GramMatrix module
jcjohnson · Oct 21, 2016 · d016c39 · d016c39
2 parents 616746b + f0b09dc
commit d016c39
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/fast_neural_style/GramMatrix.lua b/fast_neural_style/GramMatrix.lua
@@ -18,7 +18,12 @@ Output:
 
 function Gram:__init(normalize)
   parent.__init(self)
-  self.normalize = normalize or true
+  if normalize ~= nil then
+    assert(type(normalize) == 'boolean', 'normalize has to be true/false')
+    self.normalize = normalize
+  else
+    self.normalize = true
+  end
   self.buffer = torch.Tensor()
 end