Brian Broll
edeb783547
* Added basic custom layer editor. Fixes #1 * Added custom layer editing. Fixes #156 WIP #156 Updated rootviz component config WIP #156 Added template creation from parent if no code WIP #156 Added root viz panel. Changed ordering WIP #156 Added some support for 'create layer' button WIP #156 Fixed create custom layer type button WIP #156 Fixed bug with checking node for attribute WIP #156. Added support for updating meta from constructor args WIP #156 WIP #156 Added seeds WIP #156 Added attr inheritance in layer def WIP #156 Removed extra TODO * Added custom layer support to executions. Fixes #296 WIP #296 Fixed execution support * Updated GenArch for custom layers. Fixes #157 WIP #157 Updated 'standalone' default to true WIP #157 Fixed code-climate issues WIP #157 Fixed bad default WIP Removed outdated tests
28 linhas
834 B
Lua
28 linhas
834 B
Lua
require 'nn'
|
|
|
|
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
|