fix for latest torch

torch · Aug 22, 2017 · 83a5f17 · 83a5f17
1 parent e10e11d
commit 83a5f17
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 39 deletions.
diff --git a/PrintSize.lua b/PrintSize.lua
diff --git a/SequencerCriterion.lua b/SequencerCriterion.lua
@@ -53,7 +53,7 @@ function SequencerCriterion:updateGradInput(input, target)
    end
 
    if self.sizeAverage then
-      nn.utils.recursiveDiv(tableGradInput[i], seqlen)
+      nn.utils.recursiveDiv(tableGradInput, seqlen)
    end
 
    if torch.isTensor(input) then

diff --git a/init.lua b/init.lua
@@ -14,15 +14,16 @@ function nn.require(packagename)
    assert(torch.type(packagename) == 'string')
    local success, message = pcall(function() require(packagename) end)
    if not success then
-      print("missing package "..packagename..": run 'luarocks install nnx'")
+      print("missing package "..packagename..": run 'luarocks install '"..packagename.."'")
       error(message)
    end
 end
 
 
 -- c lib:
 require "paths"
-paths.require 'librnn'
+pcall(function() paths.require 'librnn' end) -- Not sure why this works...
+pcall(function() paths.require 'librnn' end)
 
 unpack = unpack or table.unpack