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
324 linhas
5.1 KiB
YAML
324 linhas
5.1 KiB
YAML
- type: SpatialMaxPooling
|
|
id: /f/0
|
|
next:
|
|
- /f/y
|
|
attributes:
|
|
padW: ''
|
|
padH: ''
|
|
ctor_arg_order: 'kW,kH,dW,dH,padW,padH'
|
|
dH: 2
|
|
dW: 2
|
|
kH: 2
|
|
kW: 2
|
|
- type: View
|
|
id: /f/1V
|
|
next:
|
|
- /f/R
|
|
attributes:
|
|
ctor_arg_order: params
|
|
params: 25088
|
|
- type: Linear
|
|
id: /f/3
|
|
next:
|
|
- /f/vo
|
|
attributes:
|
|
bias: ''
|
|
ctor_arg_order: 'inputSize,outputSize,bias'
|
|
outputSize: 1000
|
|
inputSize: 4096
|
|
- type: SpatialConvolution
|
|
id: /f/4
|
|
next:
|
|
- /f/ID
|
|
attributes:
|
|
ctor_arg_order: 'nInputPlane,nOutputPlane,kW,kH,dW,dH,padW,padH'
|
|
padH: 1
|
|
padW: 1
|
|
dH: 1
|
|
dW: 1
|
|
kH: 3
|
|
kW: 3
|
|
nOutputPlane: 512
|
|
nInputPlane: 512
|
|
- type: BatchNormalization
|
|
id: /f/4R
|
|
next:
|
|
- /f/V
|
|
attributes:
|
|
momentum: ''
|
|
affine: ''
|
|
ctor_arg_order: 'nOutput,eps,momentum,affine'
|
|
eps: 0.001
|
|
nOutput: 4096
|
|
- type: Threshold
|
|
id: /f/6
|
|
next:
|
|
- /f/J
|
|
attributes:
|
|
ip: ''
|
|
ctor_arg_order: 'th,v,ip'
|
|
v: 0.000001
|
|
th: 0
|
|
- type: SpatialConvolution
|
|
id: /f/C
|
|
next:
|
|
- /f/e
|
|
attributes:
|
|
ctor_arg_order: 'nInputPlane,nOutputPlane,kW,kH,dW,dH,padW,padH'
|
|
padH: 1
|
|
padW: 1
|
|
dH: 1
|
|
dW: 1
|
|
kH: 3
|
|
kW: 3
|
|
nOutputPlane: 512
|
|
nInputPlane: 256
|
|
- type: ReLU
|
|
id: /f/D
|
|
next:
|
|
- /f/X
|
|
attributes:
|
|
ctor_arg_order: p
|
|
p: true
|
|
- type: SpatialConvolution
|
|
id: /f/FN
|
|
next:
|
|
- /f/b
|
|
attributes:
|
|
ctor_arg_order: 'nInputPlane,nOutputPlane,kW,kH,dW,dH,padW,padH'
|
|
padH: 1
|
|
padW: 1
|
|
dH: 1
|
|
dW: 1
|
|
kH: 3
|
|
kW: 3
|
|
nOutputPlane: 512
|
|
nInputPlane: 512
|
|
- type: ReLU
|
|
id: /f/ID
|
|
next:
|
|
- /f/g
|
|
attributes:
|
|
ctor_arg_order: p
|
|
p: true
|
|
- type: BatchNormalization
|
|
id: /f/J
|
|
next:
|
|
- /f/eZ
|
|
attributes:
|
|
momentum: ''
|
|
affine: ''
|
|
ctor_arg_order: 'nOutput,eps,momentum,affine'
|
|
eps: 0.001
|
|
nOutput: 4096
|
|
- type: ReLU
|
|
id: /f/N
|
|
next:
|
|
- /f/0
|
|
attributes:
|
|
ctor_arg_order: p
|
|
p: true
|
|
- type: Linear
|
|
id: /f/R
|
|
next:
|
|
- /f/i
|
|
attributes:
|
|
bias: ''
|
|
ctor_arg_order: 'inputSize,outputSize,bias'
|
|
outputSize: 4096
|
|
inputSize: 25088
|
|
- type: SpatialMaxPooling
|
|
id: /f/U
|
|
next:
|
|
- /f/FN
|
|
attributes:
|
|
padW: ''
|
|
padH: ''
|
|
ctor_arg_order: 'kW,kH,dW,dH,padW,padH'
|
|
dH: 2
|
|
dW: 2
|
|
kH: 2
|
|
kW: 2
|
|
- type: Dropout
|
|
id: /f/V
|
|
next:
|
|
- /f/s
|
|
attributes:
|
|
v1: ''
|
|
inplace: ''
|
|
ctor_arg_order: 'p,v1,inplace'
|
|
p: 0.5
|
|
- type: SpatialConvolution
|
|
id: /f/W
|
|
next:
|
|
- /f/l
|
|
attributes:
|
|
ctor_arg_order: 'nInputPlane,nOutputPlane,kW,kH,dW,dH,padW,padH'
|
|
padH: 1
|
|
padW: 1
|
|
dH: 1
|
|
dW: 1
|
|
kH: 3
|
|
kW: 3
|
|
nOutputPlane: 512
|
|
nInputPlane: 512
|
|
- type: SpatialConvolution
|
|
id: /f/X
|
|
next:
|
|
- /f/j
|
|
attributes:
|
|
ctor_arg_order: 'nInputPlane,nOutputPlane,kW,kH,dW,dH,padW,padH'
|
|
padH: 1
|
|
padW: 1
|
|
dH: 1
|
|
dW: 1
|
|
kH: 3
|
|
kW: 3
|
|
nOutputPlane: 256
|
|
nInputPlane: 256
|
|
- type: ReLU
|
|
id: /f/Y
|
|
next:
|
|
- /f/q
|
|
attributes:
|
|
ctor_arg_order: p
|
|
p: true
|
|
- type: SpatialMaxPooling
|
|
id: /f/ZV
|
|
next:
|
|
- /f/C
|
|
attributes:
|
|
padW: ''
|
|
padH: ''
|
|
ctor_arg_order: 'kW,kH,dW,dH,padW,padH'
|
|
dH: 2
|
|
dW: 2
|
|
kH: 2
|
|
kW: 2
|
|
- type: ReLU
|
|
id: /f/b
|
|
next:
|
|
- /f/4
|
|
attributes:
|
|
ctor_arg_order: p
|
|
p: true
|
|
- type: ReLU
|
|
id: /f/e
|
|
next:
|
|
- /f/W
|
|
attributes:
|
|
ctor_arg_order: p
|
|
p: true
|
|
- type: Dropout
|
|
id: /f/eZ
|
|
next:
|
|
- /f/3
|
|
attributes:
|
|
v1: ''
|
|
inplace: ''
|
|
ctor_arg_order: 'p,v1,inplace'
|
|
p: 0.5
|
|
- type: SpatialMaxPooling
|
|
id: /f/g
|
|
next:
|
|
- /f/1V
|
|
attributes:
|
|
padW: ''
|
|
padH: ''
|
|
ctor_arg_order: 'kW,kH,dW,dH,padW,padH'
|
|
dH: 2
|
|
dW: 2
|
|
kH: 2
|
|
kW: 2
|
|
- type: Threshold
|
|
id: /f/i
|
|
next:
|
|
- /f/4R
|
|
attributes:
|
|
ip: ''
|
|
ctor_arg_order: 'th,v,ip'
|
|
v: 0.000001
|
|
th: 0
|
|
- type: ReLU
|
|
id: /f/j
|
|
next:
|
|
- /f/ZV
|
|
attributes:
|
|
ctor_arg_order: p
|
|
p: true
|
|
- type: ReLU
|
|
id: /f/l
|
|
next:
|
|
- /f/U
|
|
attributes:
|
|
ctor_arg_order: p
|
|
p: true
|
|
- type: SpatialConvolution
|
|
id: /f/n
|
|
next:
|
|
- /f/D
|
|
attributes:
|
|
ctor_arg_order: 'nInputPlane,nOutputPlane,kW,kH,dW,dH,padW,padH'
|
|
padH: 1
|
|
padW: 1
|
|
dH: 1
|
|
dW: 1
|
|
kH: 3
|
|
kW: 3
|
|
nOutputPlane: 256
|
|
nInputPlane: 128
|
|
- type: SpatialMaxPooling
|
|
id: /f/q
|
|
next:
|
|
- /f/n
|
|
attributes:
|
|
padW: ''
|
|
padH: ''
|
|
ctor_arg_order: 'kW,kH,dW,dH,padW,padH'
|
|
dH: 2
|
|
dW: 2
|
|
kH: 2
|
|
kW: 2
|
|
- type: Linear
|
|
id: /f/s
|
|
next:
|
|
- /f/6
|
|
attributes:
|
|
bias: ''
|
|
ctor_arg_order: 'inputSize,outputSize,bias'
|
|
outputSize: 4096
|
|
inputSize: 4096
|
|
- type: LogSoftMax
|
|
id: /f/vo
|
|
next: []
|
|
attributes:
|
|
ctor_arg_order: ''
|
|
- type: SpatialConvolution
|
|
id: /f/w
|
|
next:
|
|
- /f/N
|
|
attributes:
|
|
ctor_arg_order: 'nInputPlane,nOutputPlane,kW,kH,dW,dH,padW,padH'
|
|
padH: 1
|
|
padW: 1
|
|
dH: 1
|
|
dW: 1
|
|
kH: 3
|
|
kW: 3
|
|
nOutputPlane: 64
|
|
nInputPlane: 3
|
|
- type: SpatialConvolution
|
|
id: /f/y
|
|
next:
|
|
- /f/Y
|
|
attributes:
|
|
ctor_arg_order: 'nInputPlane,nOutputPlane,kW,kH,dW,dH,padW,padH'
|
|
padH: 1
|
|
padW: 1
|
|
dH: 1
|
|
dW: 1
|
|
kH: 3
|
|
kW: 3
|
|
nOutputPlane: 128
|
|
nInputPlane: 64
|