CifarNetwork.emadl 432 Bytes
Newer Older
1
2
3
4
5
6
7
8
package simpleCifar10;

component CifarNetwork<Z(2:oo) classes = 10>{
    ports in Z(0:255)^{3, 32, 32} data,
         out Q(0:1)^{classes} softmax;

    implementation CNN {
        data ->
9
10
        Convolution(kernel=(5,5), channels=8) ->
        Convolution(kernel=(5,5), channels=8) ->
11
12
13
14
15
16
17
18
        FullyConnected(units=128) ->
        Dropout()->
        FullyConnected(units=classes) ->
        Softmax() ->
        softmax

    }
}