Skip to content

Commit

Permalink
Update model.lua
Browse files Browse the repository at this point in the history
  • Loading branch information
da03 committed Dec 19, 2016
1 parent 89dad64 commit b741c93
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/model.lua
Expand Up @@ -376,7 +376,7 @@ function model:step(inputBatch, isForwardOnly, beamSize)
-- cnn
cnnGrad = cnnGrad:split(1, 1)
for i = 1, #cnnGrad do
cnnGrad[i] = cnnGrad[i]:contiguous():view(batchSize, featureMapWidth, -1)
cnnGrad[i] = cnnGrad[i]:contiguous():view(batchSize, featureMapWidth, -1)
end
self.cnn:backward(images, cnnGrad)
collectgarbage()
Expand Down

0 comments on commit b741c93

Please sign in to comment.