From b01a9e15d669656d42d6901afb771f241c3f103f Mon Sep 17 00:00:00 2001 From: Jonas Rauber Date: Fri, 8 Jan 2016 17:29:46 +0100 Subject: [PATCH] fixed typo --- neural_style.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/neural_style.lua b/neural_style.lua index e9d08d3..a657708 100644 --- a/neural_style.lua +++ b/neural_style.lua @@ -232,7 +232,7 @@ local function main(params) for i=1,#net.modules do local module = net.modules[i] if torch.type(module) == 'nn.SpatialConvolutionMM' then - -- remote these, not used, but uses gpu memory + -- remove these, not used, but uses gpu memory module.gradWeight = nil module.gradBias = nil end