Arquivos
Brian Broll a3cfa9f8e3 Added concat support to GenArch. Fixes #61 (#273)
WIP #61 Started w/ concat support

WIP #61 Added basic concat support

WIP #61 Added quick merge-fork fn-ality

WIP #64 Added nested concat support

WIP #61 Added concat tests

WIP #61 Added `npm run watch-test` cmd

WIP #61 Fixed lint errors. Removed debugger statement
2016-06-13 10:53:52 -05:00

25 linhas
440 B
Lua

require 'nn'
local net = nn.Sequential()
net:add(nn.Reshape(100))
local net_2 = nn.Sequential()
net_2:add(nn.Linear(100, 150))
net_2:add(nn.Tanh())
net_2:add(nn.Linear(150, 50))
local net_3 = nn.Sequential()
net_3:add(nn.Linear(100, 150))
net_3:add(nn.Tanh())
net_3:add(nn.Linear(150, 30))
local concat_7 = nn.Concat(1)
concat_7:add(net_3)
concat_7:add(net_2)
net:add(concat_7)
net:add(nn.Tanh())
net:add(nn.Linear(80, 7))
return net