Brian Broll
5ddeb6f331
WIP #654 Added LayerDecorator WIP #654 WIP #654 Added support for adding/removing layers WIP #654 Fixed the text for the layers WIP #654 Refactored content text WIP #654 Only delete target if it's a child WIP #654 Updated for layers-as-args WIP #654 Made a better name for layer args WIP #654 Added explicit arg types WIP #654 Changed argindex -> ctor_arg_order WIP #654 Added support for layers-as-args in GenArch WIP #654 Added import fn-ality WIP #654 Updated visualizers for layer support WIP #654 Fixed code climate issues WIP #654 Fixed setter detection WIP #654 Updated tests WIP #654 Updated tests
28 linhas
847 B
Lua
28 linhas
847 B
Lua
require 'nn'
|
|
require 'rnn'
|
|
|
|
local net = nn.Sequential()
|
|
net:add(nn.SpatialConvolution(3, 96, 11, 11, 4, 4))
|
|
net:add(nn.ReLU(true))
|
|
net:add(nn.SpatialMaxPooling(2, 2, 2, 2))
|
|
net:add(nn.SpatialConvolution(96, 256, 5, 5, 1, 1))
|
|
net:add(nn.ReLU(true))
|
|
net:add(nn.SpatialMaxPooling(2, 2, 2, 2))
|
|
net:add(nn.SpatialConvolution(256, 512, 3, 3, 1, 1, 1, 1))
|
|
net:add(nn.ReLU(true))
|
|
net:add(nn.SpatialConvolution(512, 1024, 3, 3, 1, 1, 1, 1))
|
|
net:add(nn.ReLU(true))
|
|
net:add(nn.SpatialConvolution(1024, 1024, 3, 3, 1, 1, 1, 1))
|
|
net:add(nn.ReLU(true))
|
|
net:add(nn.SpatialMaxPooling(2, 2, 2, 2))
|
|
net:add(nn.View())
|
|
net:add(nn.Dropout(0.5))
|
|
net:add(nn.Linear(25600, 3072))
|
|
net:add(nn.Threshold(0, 0.000001))
|
|
net:add(nn.Dropout(0.5))
|
|
net:add(nn.Linear(3072, 4096))
|
|
net:add(nn.Threshold(0, 0.000001))
|
|
net:add(nn.Linear(4096, 7))
|
|
net:add(nn.LogSoftMax())
|
|
|
|
return net |