AbstractRecurrent:apply() fix

torch · Jun 7, 2017 · cb227b4 · cb227b4
1 parent fee152c
commit cb227b4
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/AbstractRecurrent.lua b/AbstractRecurrent.lua
@@ -229,6 +229,12 @@ function AbstractRecurrent:evaluate()
    end)
 end
 
+function AbstractRecurrent:apply(callback)
+   for k,stepmodule in pairs(self.sharedClones) do
+      stepmodule:apply(callback)
+   end
+end
+
 function AbstractRecurrent:reinforce(reward)
    local a = torch.Timer()
    if torch.type(reward) == 'table' then

diff --git a/scripts/evaluate-rnnlm.lua b/scripts/evaluate-rnnlm.lua
@@ -1,5 +1,6 @@
 require 'nngraph'
 require 'rnn'
+require 'optim'
 local dl = require 'dataload'
 
 

diff --git a/test/test.lua b/test/test.lua
@@ -4348,8 +4348,8 @@ function rnntest.RecLSTM_maskzero()
    seqlstm2.bias:copy(reclstm.modules[1].bias)
 
    local input = torch.randn(T, N, D)
-   input[{2,1}]:fill(0)
-   input[{3,2}]:fill(0)
+   --input[{2,1}]:fill(0)
+   --input[{3,2}]:fill(0)
    local gradOutput = torch.randn(T, N, H)
 
    local zeroMask = torch.ByteTensor(T, N):zero()