Syntax fix

parent 2da70ed9
Pipeline #109888 failed with stage
in 2 minutes and 40 seconds
...@@ -6,7 +6,7 @@ component CifarNetwork<Z(2:oo) classes = 10>{ ...@@ -6,7 +6,7 @@ component CifarNetwork<Z(2:oo) classes = 10>{
implementation CNN { implementation CNN {
def conv(kernel, channels, stride=1, act=true){ def conv(kernel, channels, stride=1, act=true){
Convolution(kernel=(kernel,kernel), channels=channels, stride=(stride,stride)) -> Convolution(kernel=(kernel,kernel), channels=channels, stride=(stride,stride))
} }
def resLayer(channels, stride=1, addSkipConv=false){ def resLayer(channels, stride=1, addSkipConv=false){
( (
...@@ -15,7 +15,7 @@ component CifarNetwork<Z(2:oo) classes = 10>{ ...@@ -15,7 +15,7 @@ component CifarNetwork<Z(2:oo) classes = 10>{
| |
conv(kernel=1, channels=channels, stride=stride, act=false, ? = addSkipConv) conv(kernel=1, channels=channels, stride=stride, act=false, ? = addSkipConv)
) -> ) ->
Add() -> Add()
} }
data -> data ->
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment