README: wrap layer in MNIST example

This commit is contained in:
sorki 2023-12-08 20:04:57 +01:00 committed by Erik de Castro Lopo
parent d5db3d0b65
commit 766b962341

View File

@ -26,7 +26,8 @@ type MNIST
'[ Convolution 1 10 5 5 1 1, Pooling 2 2 2 2, Relu
, Convolution 10 16 5 5 1 1, Pooling 2 2 2 2, Reshape, Relu
, FullyConnected 256 80, Logit, FullyConnected 80 10, Logit]
'[ 'D2 28 28, 'D3 24 24 10, 'D3 12 12 10, 'D3 12 12 10
'[ 'D2 28 28
, 'D3 24 24 10, 'D3 12 12 10 , 'D3 12 12 10
, 'D3 8 8 16, 'D3 4 4 16, 'D1 256, 'D1 256
, 'D1 80, 'D1 80, 'D1 10, 'D1 10]