diff --git a/README.md b/README.md index 03608ab..f0dc278 100644 --- a/README.md +++ b/README.md @@ -24,7 +24,7 @@ specified and initialised with random weights in a few lines of code with type MNIST = Network '[ 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, FlattenLayer, 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 , 'D3 8 8 16, 'D3 4 4 16, 'D1 256, 'D1 256