Brian Broll
822a4080e5
WIP #40 Updated overfeat devTest Fixed setting values to null. WIP #40 WIP #40 Fixed attr name typos WIP #40 Updated tests
10 linhas
159 B
Lua
10 linhas
159 B
Lua
require 'nn'
|
|
|
|
local net = nn.Sequential()
|
|
net:add(nn.Reshape(100))
|
|
net:add(nn.Linear(100, 300))
|
|
net:add(nn.HardTanh())
|
|
net:add(nn.Linear(300, 10))
|
|
|
|
return net
|