allenai / forces Goto Github PK

require 'nn'
function rnn(inDim, hidDim, outDim , rho, shared)

     local function SelectRange(m,n)
     	 local C = nn.ConcatTable()
     	 for i= m,n do
     	    C:add(nn.SelectTable(i))
     	 end
     	 return C
     end

     local function JoinTable_firstTwo(dim)
     	 local C = SelectRange(1,2)
     	 return nn.Sequential():add(C):add(nn.JoinTable(dim))
     end

     local function joinUnit()  
     	return nn.Sequential():add(JoinTable_firstTwo(2)):add(nn.Linear(inDim+hidDim,hidDim)):add(nn.ReLU(true))
     end
     local function maxUnit() 
       return nn.Sequential():add(nn.SelectTable(2)):add(nn.Linear(hidDim,outDim))
     end

     local rnnUnit = nn.Sequential()
     for i=1,rho do
         hidUnit = nn.ConcatTable()
                      :add(nn.SelectTable(1))
                      :add(joinUnit())
                      :add(maxUnit())
        if i>1 then
        	for j=1,i-1 do
        		hidUnit:add(nn.SelectTable(2+j))
        	end     
        end 
        rnnUnit:add(hidUnit)
     end
	 rnnUnit:add(SelectRange(3,2+rho))
	 
	 if shared == true then	 	
	 	local linUnits = rnnUnit:findModules('nn.Linear')
	 	for i=3,#linUnits,2 do
        	linUnits[i] = linUnits[1];
         	linUnits[i+1] =linUnits[2]    
     	end
     end

	 return rnnUnit
end
config ={}
config.force_size = 4096
config.rho = 6
config.RNNhSize = 1000;
config.RNNoSize = 18;
config.nCategories = 18;
local start = nn.ConcatTable()
                    :add(nn.Identity())
                    :add(nn.Sequential():add(nn.Linear(4096+config.force_size,config.RNNhSize)):add(nn.ReLU(true)))
local model= nn.Sequential():add(start):add(rnn(4096+config.force_size, config.RNNhSize, config.RNNoSize, config.rho, false))

local input = torch.rand(4096*2)
model:forward(input)

bad argument #2 to '?' (out of bounds at /Users/jsuit/torch/pkg/torch/lib/TH/generic/THStorage.c:202)
stack traceback:
	[C]: at 0x04526e60
	[C]: in function '__index'
	/lua/5.1/nn/JoinTable.lua:30: in function 
<lua/5.1/nn/JoinTable.lua:21>
	[C]: in function 'xpcall'
       /5.1/nn/Container.lua:63: in function 'rethrowErrors'
	/lua/5.1/nn/Sequential.lua:44: in function 
</lua/5.1/nn/Sequential.lua:41>
	[C]: in function 'xpcall'
	/lua/5.1/nn/Container.lua:63: in function 'rethrowErrors'
	/lua/5.1/nn/Sequential.lua:44: in function <lua/5.1/nn/Sequential.lua:41>
	[C]: in function 'xpcall'
	lua/5.1/nn/Container.lua:63: in function 'rethrowErrors'
	/lua/5.1/nn/ConcatTable.lua:11: in function 
<lua/5.1/nn/ConcatTable.lua:9>
	[C]: in function 'xpcall'
	/lua/5.1/nn/Container.lua:63: in function 'rethrowErrors'
	lua/5.1/nn/Sequential.lua:44: in function 'forward'
	/rnnUnit.lua:61: in main chunk
	[C]: in function 'dofile'
	install/lib/luarocks/rocks/trepl/scm-1/bin/th:145: in main chunk
	[C]: at 0x01041ff1c0